[프로그래머스] 소수 만들기 - 파이썬

Programmers

소수 만들기

https://school.programmers.co.kr/learn/courses/30/lessons/12977?language=python3

프로그래머스

문제

코드

• combinations를 활용하여 조합의 합을 계산하는 방법
• for문을 3개 돌리는 것보다는 효율적인 방법

from itertools import combinations

def solution(nums):
    answer = 0
    for combo in combinations(nums, 3):
        n = sum(combo)
        for i in range(2, int(n**(0.5))+1):
            if n % i == 0:
                ab = False
                break
            else:
                ab = True
        if ab is True:
            answer += 1
    return answer