[백준/소수] 19699번 소-난다!

https://www.acmicpc.net/problem/19699

19699번: 소-난다!

 

 

itertools의 combination을 사용해서, 조합을 구하고

그 sum이 소수인 것을 확인한다.

 

import sys
import math
import itertools

input = sys.stdin.readline

n,m = map(int,input().split())
l = list(map(int,input().split()))
l.sort()

answer = set()


def isPrime(n):
    if n<=1: return False
    if (n%2==0):
        if n==2: return True
        else: return False

    for i in range(3,int(math.sqrt(n))+1,2):
        if(n%i==0): return False

    return True


for i in itertools.combinations(l,m):

    x = sum(i)
    if isPrime(x):
        answer.add(x)
answer = list(answer)
answer.sort()

if len(answer)==0:
    print(-1)
else:
    for i in answer:
        print(i, end = ' ')