✅ 코드
from itertools import combinations
import math
def is_prime(x):
for i in range(2, int(math.sqrt(x)) + 1):
if x % i == 0:
return False
return True
def solution(nums):
answer = 0
combis = list(combinations(nums, 3))
for combi in combis:
if is_prime(sum(combi)):
answer += 1
return answer
주어진 배열에서 3개의 수를 조합해 더하면 소수가 되는 케이스의 수를 구해야 합니다.
다행스럽게도(?) 파이썬의 itertools 모듈을 사용하면 수들의 조합을 구할 수 있습니다.
소수 여부를 판별하는 is_prime 함수를 별도로 선언해 문제를 해결했습니다.
'알고리즘' 카테고리의 다른 글
[파이썬]프로그래머스 Lv.1 완주하지못한선수 (0) | 2023.06.03 |
---|---|
[파이썬]프로그래머스 Lv.1 다트게임 (0) | 2023.06.02 |
[파이썬]프로그래머스 Lv.1 같은숫자는싫어 (0) | 2023.06.01 |
[파이썬]프로그래머스 Lv.1 달리기경주 (0) | 2023.05.31 |
[파이썬]프로그래머스 Lv.1 추억점수 (0) | 2023.05.30 |