Haribo ML, AI, MATH, Algorithm

소수 만들기


소수 만들기

from itertools import combinations

def prime_number(x):
    answer = 0
    for i in range(1,int(x**0.5)+1):
        if x%i==0:
            answer+=1
    return 1 if answer==1 else 0

def solution(nums):
    return sum(prime_number(sum(c)) for c in combinations(nums,3))

하…sympy모듈 사용 가능했으면 정말 간단했을 텐데 아쉽다. sympy

from itertools import combinations
from sympy is isprime
def solution(nums):
    return sum(isprime(sum(c)) for c in combinations(nums,3))

Similar Posts

이전 포스트 [3차] 파일명 정렬

다음 포스트 영어 끝말잇기

Comments