[Programmers][Python][Level 1] 소수 만들기

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

 

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

 

문제 

주어진 숫자 중 3개의 수를 더했을 때 소수가 되는 경우의 개수를 구하려고 합니다.

숫자들이 들어있는 배열 nums가 매개변수로 주어질 때, nums에 있는 숫자들 중 서로 다른 3개를 골라 더했을 때 소수가 되는 경우의 개수를 return 하도록 solution 함수를 완성해주세요.

 

제한 조건

• nums에 들어있는 숫자의 개수는 3개 이상 50개 이하입니다.
• nums의 각 원소는 1 이상 1,000 이하의 자연수이며, 중복된 숫자가 들어있지 않습니다.

 

입출력 예

num	result
[1,2,3,4]	1
[1,2,7,6,4]	4

 

입출력 예 설명

입출력 예 #1
[1,2,4]를 이용해서 7을 만들 수 있습니다.

 

입출력 예 #2
[1,2,4]를 이용해서 7을 만들 수 있습니다.
[1,4,6]을 이용해서 11을 만들 수 있습니다.
[2,4,7]을 이용해서 13을 만들 수 있습니다.
[4,6,7]을 이용해서 17을 만들 수 있습니다.

 

풀이

다음과 같이 문제를 접근했습니다.

 

1. 서로 다른 3개의 숫자를 반복문 또는 조합을 통해서 구해서 더하기

2. 3개의 숫자를 더한 합이 소수인지 판별하기

 

첫 번째 풀이

1. 서로 다른 숫자 3개를 더해주기 위해서 저는 python의 combinations를 사용했습니다.
   
   • combination() : 중복을 허용하지 않고 순서 의미가 있는 조합을 리턴해줍니다.
2. is_prime_number(number) 함수를 정의해서 소수를 판별해서 소수이면 True를 리턴했습니다.

from itertools import combinations

def solution(nums):
    answer = 0
    
    combi = list(combinations(nums, 3))
    check = False
    
    for i,j,k in combi:
        total = i + j + k
        check = prime_number(total)
        if check == True:
            answer += 1
              
    return answer

def is_prime_number(number):    
    if number != 1:                 
        for f in range(2, number):  
            if number % f == 0:     
                return False    
    else:                        
        return False            
    return True

 

두 번째 풀이

소수를 판별하는 부분은 위와 동일하지만 서로 다른 3개를 구해서 더하는 방법은 3중 for문을 사용했습니다.

def solution(nums):
    answer = 0
    check = False

    for i in range(len(nums) - 2): 
        for j in range(i+1, len(nums) - 1):
            for k in range(j+1, len(nums)): 
                check = is_prime_number(nums[i]+nums[j]+nums[k])
                if check == True:
                    answer += 1
                
    return answer

def is_prime_number(number):
    if number != 1:
        for i in range(2, number):
            if number % i  == 0:
                return False
    else:
        return False
        
    return True