파이썬 알고리즘-113 (프로그래머스) 소수 만들기

jiffydev·2021년 2월 19일
0

Algorithm

목록 보기
120/134

코드

from itertools import combinations
def solution(nums):
    answer = 0
    # 3개를 골라 더한 값들의 조합 & 리스트를 lst 변수에 저장
    lst=list(map(sum,list(combinations(nums,3))))
    for n in lst:
        # n이 소수인지 판별하기 위해 2부터 n의 제곱근까지의 리스트 생성
        n_lst=[i for i in range(2,int((n**0.5))+1)]
        for j in n_lst:
            if n%j==0:
                break
        else:
            answer+=1
    return answer

다른 사람의 풀이

def solution(nums):
    from itertools import combinations as cb
    answer = 0
    for a in cb(nums, 3):
        cand = sum(a)
        for j in range(2, cand):
            if cand%j==0:
                break
        else:
            answer += 1
    return answer
profile
잘 & 열심히 살고싶은 개발자

0개의 댓글