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

from itertools import combinations

def solution(nums):
	cnt = 0
	arr = list(map(sum, combinations(nums, 3)))
	
	for i in arr:
		flag = 0
		for j in range(2, int(i ** 0.5)+1):
			if i % j == 0:
				flag = 1
				break

		if flag == 0:
			cnt += 1
	return cnt