PTA 选数

7-5 选数

已知n个整数x1,x2,x3...xi，以及1个整数k(k<n)。从 n 个整数中任选 k个整数相加，可分别得到一系列的和。例如当 n=4，k=3，4个整数分别为3,7,12,19 时，可得全部的组合与它们的和为：
3+7+12=22，
3+7+19=29，
7+12+19=38，
3+12+19=34，

现在，要求你计算出和为素数共有多少种。

例如上例，只有一种的和为素数：3+7+19=29

输入格式:

第一行两个空格隔开的整数 n,k（1≤n≤20，k<n）
第二行n个整数，两数之间空格隔开（1≤xi≤1000000）

输出格式:

输出一个整数，表示种类数。

输入样例:

在这里给出一组输入。例如：

4 3
3 7 12 19

输出样例:

在这里给出相应的输出。例如：

1

代码长度限制

16 KB

时间限制

1000 ms

内存限制

128 MB

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=50;
int a[maxn];
int n,k,ans;
bool prime(int x){//判断是不是素数
	for(int i=2;i<=sqrt(x);i++){
		if(x%i==0)return false;
	}
	return true;
}
void dfs(int t,int sum,int d){
    if(t==k){
        if(prime(sum))ans++;//达到k个 检验是否是素数 计数
    }
    else{
        for(int i=d;i<n;i++){//没到k个  继续从后边找下一个数
            dfs(t+1,sum+a[i],i+1);
        }
    }
}
int main(){
    cin>>n>>k;
    for(int i=0;i<n;i++){
    cin>>a[i];
}
    dfs(0,0,0);
    cout<<ans<<endl;
    return 0;
}