题目描述:已知 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)
输出:
满足条件的种数
例:
输入
4 3
3 7 12 19
输出
1
#include <iostream>
#include <cmath>
using namespace std;
int n,k,sum=0,t,a[20];
bool is_prime(long long n) //判断是否为素数
{
for(long long int i=2;i<=sqrt(n);i++)
{
if(n%i==0)
return false;
}
return true;
}
void dfs(int x,int y) //x个数相加,第y个数
{
int i;
if(x==k) //有k个数相加
{
if(is_prime(sum)) //判断和是否为素数
t++; //素数则加一
return ; //返回上一层
}
for(i=y+1;i<=n;i++)
{
sum+=a[i]; //加入元素
dfs(x+1,i); //多加一个数
sum-=a[i]; //去掉元素
}
}
int main() {
cin>>n>>k;
for(int i=1;i<=n;i++)
cin>>a[i];
dfs(0,0);
cout<<t;
return 0;
}