파이썬/백준

프로그래머스 _ sqrt와 조합 사용한 문제

채린.__. 2023. 4. 30. 03:07

from itertools import combinations
import math

def is_prime(n):
	if n<2:
		return False
	for i in range(2, int(math.sqrt(n))+1):
		if n%i ==0:
			return False
		return True

def solution(numbs):
	number=0
	for a,b,c in combinations(numbs,3):
		if is_prime(a+b+c) == True:
			number+=1
		else:
			continue
	return number