[ 프로그래머스 / PYTHON ] 소수 만들기

문제 설명

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

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

코드 설명

1. 소수를 입력한 리스트 작성
2. 3중 for문 돌면서 합이 소수면 count ++

1+2+4 와 1+4+2는 다른 것 주의. 정답률이 23퍼 언저리가 나온다면 중복처리를 해줘서 그럼.

2021 8월 코드

def solution(nums):
    answer = 0
    li = [True] *3000
    li[0] = False
    li[1] = False
    result = []
    #소수 리스트 만들기
    int_=int(3000**0.5)
    for i in range(2, int_):
        if li[i] == True:
            for j in range(i + i,3000,i):
                li[j] = False
    prime = [i for i in range(3000) if li[i] == True]


    for i in nums:
        for j in nums:
            if i<j:
                for k in nums:
                    if j<k:
                        if i+j+k in prime:
                            answer += 1
    return answer

2022.01.07

def solution(nums):
    count = 0
    prime = [0] * 3000
    for j in range(2, 3000//2):
        if prime[j] == 0:
            for k in range(j + j,3000,j):
                prime[k] = 1

    for i in range(len(nums)):
        for j in range(i+1, len(nums)):
            for k in range(j+1, len(nums)):
                temp = nums[i] + nums[j] + nums[k]
                if prime[temp] == 0 :
                    count += 1
                
    return count