[Python] Summer/Winter Coding(~2018) : 소수 만들기

코딩테스트 연습 - 소수 만들기

 

💡문제💡

숫자들이 들어 있는 배열 num가 주어졌을 때, 주어진 숫자 중 3개를 더해 소수가 되는 경우의 개수를 return 하시오.

 

 

[제한 사항]

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

 

 

 

코드 및 풀이👩🏻‍💻

'''

from itertools import combinations

def check(i):
    for j in range(2,i):
        if i%j==0:
            return 0
    return 1

def solution(nums):
    answer = list(combinations(nums,3))
    count = 0

    for i in range(len(answer)):
        if(check(sum(answer[i]))==1):
            count +=1
    return count

'''

*combinations() 함수 => list 내 모든 조합 구하기

ex ) combinations(nums,3) : nums 리스트 내 3개 요소를 포함하는 모든 조합 구하기