[Python] 소수 만들기

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

1. combinations 를 사용해서 nums에서 3개를 고른다.

2. 반복문 안에서 3개의 수를 더한 수를 구하고 2부터 그 수의 제곱근 까지 나누어서 나머지가 0인 수가 있으면 소수가 아님을 이용한다.

3. 만약 소수라면 answer 의 개수를 1씩 증가 시킨다.

 

from itertools import combinations as c

def solution(nums):
	answer = 0
    comb = list(c(nums,3))
	for n in comb:
    	current_n = sum(n)
        isPrime = True
        for i in range(2, int(current_n ** 0.5) + 1):
        	if current_n % i == 0:
            	isPrime = False
        if isPrime:
        	answer += 1
    return answer