from itertools import combinations
import math
def solution(nums):
answer = 0
sum_list = []
num_list = list(combinations(nums,3))
for i in num_list :
sum_list.append(sum(i))
# print(sum_list)
for i in range(len(sum_list)) :
j = 2
while j <= math.sqrt(sum_list[i]) :
if sum_list[i] % j == 0 :
# print(sum_list[i])
prime = 0
break
else :
prime = 1
j += 1
if(prime) : answer += 1
return answer
배운점 :
1. 조합을 구하는 함수 combination이 있음을 알았다.
2. 제곱근을 구하는 함수 math.sqrt() 를 배웠다.
3. 2부터 자기자신의 제곱근까지의 숫자로 나눠지지 않는 수를 소수로 정의할 수 있다.