[Programmers] Summer/Winter Coding(~2018) : 소수 만들기 (Python)

1일 1문제 70일차!

오늘의 문제는 프로그래머스의 Summer/Winter Coding(~2018)의 소수만들기입니다.

프로그래머스

Solution

from itertools import combinations

def getPrimaryNum_Eratos(N): 
    nums = [True] * (N + 1) 
    for i in range(2, len(nums) // 2 + 1): 
        if nums[i] == True: 
            for j in range(i+i, N, i): 
                nums[j] = False 
    return [i for i in range(2, N) if nums[i] == True]
    

def solution(nums):
    permu_list = []
    
    my_permu_list = list(combinations(nums, 3))
    
    for i in range(len(my_permu_list)):
        permu_list.append(sum(my_permu_list[i]))
    
    
    max_num = max(permu_list)
    
    prime_nums = getPrimaryNum_Eratos(max_num + 1)
    
    print(permu_list)
    
    answer_nums = [num for num in permu_list if num in prime_nums]
    
    print(answer_nums)
    
    answer = len(answer_nums)
    
    return answer

 

SOMJANG/CODINGTEST_PRACTICE