P1036 [NOIP2002 普及组] 选数

P1036 [NOIP2002 普及组] 选数 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

关键就俩：

怎么从n个数中任选k个数（办法很多，dfs是常见的解决方法，这里用python自带的方法方便很多）

质数判断

import itertools
n,k=map(int,input().split())
x=input().split(" ")
number=[] #存放每次任选出来的数字元组
s=[] #存放每个元组的和
cnt=0 #统计质数个数
for i in range(n):
    number.append(int(x[i]))
cmb=list(itertools.combinations(number,k))
#求和
for c in cmb:
    s.append(sum(c))
#判断质数
for i in s:
    flag=0
    if i<=2:
        cnt+=1
    else:
        for j in range(2,i):
            if i%j==0:
                flag=1
                break
        if flag==0:
            cnt+=1
print(cnt)