(Python/LV1) 소수 만들기

문제 출처:https://school.programmers.co.kr/learn/courses/30/lessons/12977

프로그래머스

풀이:

import itertools

def sosu(n):
    if n < 2:
        return 0
    else:
        for i in range(2, n):
            if n%i == 0:
                return 0
        return 1

def solution(nums):
    answer = 0
    combination = list(itertools.combinations(nums, 3))
    for i in range(len(combination)):
        answer += sosu(sum(combination[i])) # (뽑힌 3개 수 더해서 sosu 함수에 넣음)

    return answer

print(solution([1,2,3,4]))

 

itertools.combinations(nums, 3) 

>> Nums 리스트에서 3가지를 뽑는 방법