본문 바로가기
Develop/알고리즘

[백준/Python] Silver II #19699 소-난다!

by favorcat 2023. 5. 22.
반응형
 

19699번: 소-난다!

지난 번 헛간 청약의 당첨우(牛)가 발표됐다. 청약에 당첨된 소들은 날아갈 듯이 기뻐하다가 진짜로 하늘을 날았다. 하지만 이후로 소들은 날 수 없었다. 그러던 어느 날, 꿀벌에게 쏘이면 잠깐

www.acmicpc.net

문제

입력

출력

제한

풀이

from itertools import combinations

def isPrime(n):
  i = 2
  while i * i <= n:
    if n % i == 0:
      return False
    i += 1
  return True

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

c = list(combinations(h,m))
ans = []

for a in c:
  total = sum(a)
  if isPrime(total):
    ans.append(total)

ans = sorted(list(set(ans)))
if len(ans) == 0:
  print(-1)
else:
  for i in ans:
    print(i, end=" ")
반응형

Comment