[programmers] 소수 만들기

데린이·2022년 4월 8일
0

숫자 배열의 가능한 combination 중에서 소수인 숫자를 찾아라.
https://programmers.co.kr/learn/courses/30/lessons/12977
Tips.
1. for a in b 에서 b가 변동이 없는 변수로 진행할 것
2. from itertools import combinations 사용

22-04-19 code

from itertools import combinations 
import numpy as np

def solution(nums):
    result = 0
    
    comb_list = list(combinations(nums,3))
    for cl in comb_list:
        prime = sum(cl)
        prime_factor = True
        
        if prime % 2 != 0:
            for p in range(3,int(np.sqrt(prime))+1,2):
                if prime % p == 0:
                    prime_factor = False
                    break
        else:
            prime_factor = False
                    
        if prime_factor:
            result += 1
        
    return result
profile
취뽀를 기원하는 취준생입니다!

0개의 댓글