选数
已知 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)。x1,x2,⋯,xn(1≤xi≤5000000)。
输出格式为:
一个整数(满足条件的种数)。
Sample Input
4 3
3 7 12 19
Sample Output
1
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
using namespace std;
int n,k,a[100005],ans=0;
int is_prime(int x) {
for(int i=2; i<=sqrt(x); i++)
if(x%i==0) {
return 0;
}
return 1;
}
void dfs(int start ,int count,int sum) {
if(count==k&&is_prime(sum)) {
ans++;
}
for(int i=start; i<n; i++)
dfs(i+1,count+1,sum+a[i]);
}
int main() {
cin>>n>>k;
for(int i=0; i<n; i++)
cin>>a[i];
dfs(0,0,0);
cout<<ans<<endl;
return 0;
}