[Programmers] 소수 만들기

 

 

문제 

 

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

 

 

 

제한 사항

 

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

 

나의 코드 

 

def solution(nums):
    cnt = 0
    arr = []
    
    # 숫자 세 개를 더해서 나올 수 있는 모든 경우의 수를 담은 배열 만들기 
    for i in range(len(nums)):
        for t in range(i+1,len(nums)):
            for k in range(t+1,len(nums)):
                p = nums[i] + nums[t] + nums[k] 
                arr.append(p)
    
    # 1 ~ 자기자신까지의 수를 나눌 때 그 횟수가 2인 경우에는 소수이기에 cnt를 증가시킴
    for y in arr:
        idx = 0
        for z in range(1,y+1):
            if y % z == 0:
                idx += 1
        
        if idx == 2:
            cnt += 1
                   
    return cnt

분석 > 세 수를 더해서 나올 수 있는 모든 경우의 수에서 소수를 판별하는 부분에서 어려움을 겪었다. 하지만 생각해보니 소수란 자기 자신과 1로만 나눌 수 있는 수라는 정의를 활용해서 나눌 수 있는 횟수가 2인 경우만 따로 그 횟수를 구해주면 소수의 개수를 구할 수 있는 것이었다. 

 

 

 

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