[프로그래머스] 소수 만들기 (level.1)

 

문제링크

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

코딩테스트 연습 - 소수 만들기

 

나의 풀이

1. 소수임을 확인하는 check함수와, 메인함수인 solution 함수로 구성했다.
2. check에서는 2부터 반복문을 돌며 나눠보고, 나머지가 0이나오는순간 False를 출력하게 하면서 소수룰 구별했다.
3. solution에서는 combination 함수로 3개씩 조합을 이뤄놓고, sum을구해 check함수로 보냈다. 
4. check함수를 거쳐 나온 결과 True이면 answer에 +1 해주는 방식이다.

from itertools import combinations

def check(num):
    for i in range(2, num):
        if num % i == 0:
            return False
    return True
    
def solution(nums):
    answer = 0
    combi = list(combinations(nums, 3))
    for n in combi:
        if check(sum(n)) == True:
            answer +=1
    return answer

 

다른사람 풀이

대부분 combination을 활용한 코드가 좋아요를 많이 받았다. 나처럼 함수를 두개로 나눠서 푼사람도 있었고, 그냥 solution 하나의 함수로 푼 풀이도 있었다. 다 풀이가 비슷하므로 링크만 남겨두겠따.