[프로그래머스] 코딩테스트 연습 > Summer/Winter Coding(~2018) > 소수 만들기

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

프로그래머스

LV1. 소수 만들기

 

💡아이디어

주어진 숫자들로 만들수 있는 조합을 생각하고, 소수인지 확인하는 문제이다.

레벨1의 쉬운 문제인데 바보 같이 '소수'의 정의가 순간적으로 생각이 안나고 홀수로 착각하고 풀었다 바보같이 ㅎㅎ

두가지 방법으로 풀어보았는데, 첫번째는 반복문 3개를 써서 3개의 수 합을 구했고 두번째는 itertools 모듈에 combinations 함수를 사용했다. 

 

 

💻 코드작성

from itertools import combinations, permutations

def check(number):
# 소수인지 체크하는 함수
# 소수: 1과 자기 자신만을 약수로 하는 숫자
    cnt = 0
    for i in range(1, number+1):
        if number % i == 0 : cnt += 1
        
    if cnt == 2 : return True

    return False

def solution(nums):
    answer = 0
# 첫번째 풀이: 반복문 3개 사용
    # temp = 0
    # cnt = 0
    # for i in range(len(nums)-2):
    #     for j in range(i+1, len(nums)-1):
    #         for k in range(j+1, len(nums)):
    #             temp = nums[i] + nums[j] + nums[k]
    #             if check(temp) :
    #                 print(temp)
    #                 answer += 1 
    
# 두번째 풀이: itertools 모듈 combination 함수 사용
    comb_list = []
    for i in combinations(nums, 3):
         comb_list.append(sum(i))

    for i in range(len(comb_list)):
        if check(comb_list[i]):
            answer += 1
        
    return answer