프로그래머스[Lv. 1]_소수 만들기

🤷‍♂️문제설명

주어진 숫자 중 3개의 수를 더했을 때 소수가 되는 경우의 개수를 구하려고 합니다.

숫자들이 들어있는 배열 nums가 매개변수로 주어질 때, nums에 있는 숫자들 중 서로 다른 3개를 골라 더했을 때 소수가 되는 경우의 개수를 return 하도록 solution 함수를 완성해주세요.

 

🙅‍♀️제한조건

• nums에 들어있는 숫자의 개수는 3개 이상 50개 이하입니다.
• nums의 각 원소는 1 이상 1,000 이하의 자연수이며, 중복된 숫자가 들어있지 않습니다.

 

💁입출력 예

nums	result
[1,2,3,4]	1
[1,2,7,6,4]	4

입출력 예 #1
[1,2,4]를 이용해서 7을 만들 수 있습니다.

입출력 예 #2
[1,2,4]를 이용해서 7을 만들 수 있습니다.
[1,4,6]을 이용해서 11을 만들 수 있습니다.
[2,4,7]을 이용해서 13을 만들 수 있습니다.
[4,6,7]을 이용해서 17을 만들 수 있습니다.

 

🙆코드

import itertools

def solution(nums):
    sosu = set(itertools.combinations(nums, 3))
    sumi = [sum(c) for c in sosu]
    
    cnt = 0
    for j in sumi:
        for k in range(2, int(j**0.5) + 1):
            if j % k == 0:
                break
        else:
            cnt += 1
        
    return cnt

 

🗨️첨언

우선 문제에서 주어진 리스트 속 두개의 수를 골라 더하면 이라는 조건이 있기 때문에

combinations 함수를 사용할 수 있다.

그리고 set을 이용해 중복을 방지하고,

어차피 경우의 수를 구하는 것이 아닌 소수 갯수를 구하면 되기 때문이다.

그리고 조합 한 수를 더한 리스트를 만들고

그 리스트를 순회하면서 1과 자신을 제외한 약수가 있는지 찾아본다.