题目描述
已知n个整数x1,x2,……xn,以及一个整数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个数
x1 x2 ……xn(1≤xi≤5000000),各数之间用一个空格隔开)
输出格式
一个整数(满足条件的种数)。
输入输出样例
输入 #1 复制
4 3
3 7 12 19
输出 #1 复制
1
dfs解决问题 dfs dfs dfs dfs dfs
#include<bits/stdc++.h>
using namespace std;
#define N 22
#define M 100000001
int a[N], n, k, ans = 0;
bool f[M];
bool is_prime(int x){
for(int i=2; i<=sqrt(x); i++){
if(x%i == 0){
return false;
}
}
return true;
}
void dfs(int s, int c, int p){//s:已选数的和 c:已选c个数 p:已选到了数组p下标的数
if(c == k){
if(is_prime(s)) {
++ans;
}
return ;
}
for(int i=p+1; i<=n; i++){
dfs(s+a[i], c+1, i);
}
return ;
}
int main(){
cin >> n >> k;
for(int i=1; i<=n; i++)
cin >> a[i];
dfs(0,0,0);
cout << ans;
return 0;
}