코딩 테스트 (Coding Test)

[프로그래머스] 소수 만들기

Universe_lee 2023. 1. 10. 16:54

나의 답안

from itertools import combinations

def find(n):
    for i in range(2, int(n * .5) + 1):
        if n % i == 0:
            return False
    return True


def solution(nums):
    data = []
    count = 0
    
    for i in combinations(nums, 3):
        data.append(sum(i))
            
    for i in data:
        if find(i):
            count += 1
    return count