L1 : 소수 만들기 Python

jhyunn·2023년 1월 23일
0

Programmers

목록 보기
63/69

L1 : 소수 만들기 Python

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

from itertools import combinations
def solution(nums):
    answer = 0
    for p in combinations(nums, 3):
        sosu = 0
        p = sum(p)
        for i in range(2, int(p**0.5) + 1):
            if p%i==0:
                sosu = 1
                break
        if sosu == 0:
            answer += 1
    return answer

combinations, 공약수를 구할 때는 sqrt(n)까지만 확인하면 됨을 활용하는 문제

#itertools #combinations

profile
https://github.com/Sungjeonghyun

0개의 댓글