프로그래머스 Summer/Winter Coding(~2018) - 소수 만들기

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

문제 설명

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

제한사항

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

풀이

def solution(nums):
    answer = 0
    for num1 in range(len(nums)-2):
        for num2 in range(num1+1, len(nums)-1):
            for num3 in range(num2+1, len(nums)):
                isPrime = True
                sum = nums[num1] + nums[num2] + nums[num3]
                if sum % 2 == 0:
                    continue
                for i in range(int(sum/2)-1 if int(sum/2) % 2 == 0 else int(sum/2), 2, -2):
                    if sum % i == 0:
                        isPrime = False
                        break
                if isPrime:
                    answer += 1

    return answer

• 3개를 뽑아서 sum에 더한뒤
• sum을 가지고 반으로 나눈뒤 그게 짝수면 -1 해주고 홀수면 그대로 해서 i로 정해줌
• i는 2씩 줄어들면서 검사함

다른 사람의 풀이

from itertools import combinations
# 콤비네이션 함수로 3개를 뽑아온다 a에
def solution(nums):
    answer = 0
    for a in combinations(nums, 3):
        sumNum = sum(a)  # 다 더한걸 가지고
        isPrime = True
        for i in range(2, sumNum): # 하나씩 그냥 다 검사함
            if sumNum % i == 0:
                isPrime = False
                break
        if isPrime:
            answer += 1
    return answer