[Python] 프로그래머스. 소수 만들기 (Level 2)

https://programmers.co.kr/learn/courses/30/lessons/12977

프로그래머스

1. combinations으로 숫자 세 개 뽑기

2. 뽑은 세 개의 숫자를 합한 값이 소수인지 판별하기.

 

 

	from itertools import combinations
	import math
	def solution(nums):
	candidates = combinations(nums, 3)
	answer = 0
	for item in candidates:
	sums = sum(item)
	is_prime = True
	for i in range(2, int(math.sqrt(sums)) + 1):
	if sums % i == 0:
	is_prime = False
	break
	if is_prime:
	answer += 1
	return answer

view raw [프로그래머스] 소수 만들기.py hosted with ❤ by GitHub