Summer/Winter Coding(~2018) 소수 만들기

이기현·2021년 9월 30일

코딩테스트 준비

목록 보기
20/20
from itertools import combinations
import math

def solution(nums):

    answer = 0
    sum_list = []
    num_list = list(combinations(nums,3))
    
    for i in num_list :
        sum_list.append(sum(i))
  #  print(sum_list)
    
    for i in range(len(sum_list)) :
        j = 2
        while j <=  math.sqrt(sum_list[i]) :
            if sum_list[i] % j == 0 :
 #               print(sum_list[i])
                prime = 0
                break
            else :
                prime = 1
                j += 1
        if(prime) : answer += 1
    return answer

배운점 :
1. 조합을 구하는 함수 combination이 있음을 알았다.
2. 제곱근을 구하는 함수 math.sqrt() 를 배웠다.
3. 2부터 자기자신의 제곱근까지의 숫자로 나눠지지 않는 수를 소수로 정의할 수 있다.

profile
실력을 쌓아가는 하루하루

0개의 댓글