소수 만들기

문제

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

숫자들이 들어있는 배열 nums가 매개변수로 주어질 때,

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

조건

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

입출력 예

 

---

def solution(nums):
    answer = 0
    arr = []
    for i in combination(nums, 3):
        arr.append(sum(i))
    for i in range(len(arr)):
        for j in range(2,arr[i]):
            if arr[i]%j==0:
                break
        else: answer += 1
    return answer

def combination(nums, r):    #조합 제너레이터 작성

    for i in range(len(nums)):
        if r == 1:
            yield[nums[i]]
        else:
            for j in combination(nums[i+1:], r-1):
                yield[nums[i]] + j

---

코딩테스트 중 외부라이브러리를 사용하지 못하게 하는 경우가 있다고 들었다.

그래서 조합을 함수로 구현하였다.

 

제네레이터와 일반 함수의 차이는 return을 사용하냐 yield를 사용하냐인데,

return은 한 함수당 한번 실행되고, 실행되면 반환값을 반환하고 함수가 종료되지만

yield은 여러 값을 입력 할 수 있고, 입력된 값들은 next()를 통해 입력 된 값이 모두 소진될때까지 순차적으로

반환된다.