已知 n 个整数 x1,x2,⋯,xn,以及 11个整数 k(k<n)。从 n个整数中任选 k 个整数相加,可分别得到一系列的和。例如当 =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。
这个题目最难的地方在于进行搜索的过程中要注意不要重复搜索相同的元素,比如第1,4,5个数的和跟第5,4,1的和是一样的,为了避免出现这种情况,我们额外引入一个参数(记录自己上次搜索的位置)下一次的搜索起点就是上一次搜索的位置加一。例如:1的下一次搜索起点就是2,这样一层一层搜索下去就可以充分避免重复问题。
#include<iostream>
#include<cstdio>
using namespace std;
int n,k;
long long ans=0;
int q[30];
bool check(int x)//检查和是否是素数
{
for(int i=2;i*i<=x;i++)
{
if(x%i==0)
return false;
}
return true;
}
void dfs(int x,int sum,int qx)第一个变量记录加了多少个数了,第二个变量记录和,第三个变量记录上一次搜索到第几个数从而避免重复搜索
{
if(x==k)
{
if(check(sum))
ans++;
return;
}
for(int i=qx;i<n;i++)
{
dfs(x+1,sum+q[i],i+1);
}
}
int main()
{
cin>>n>>k;
for(int i=0;i<n;i++)
scanf(" %d",&q[i]);
dfs(0,0,0);
cout<<ans;
return 0;
}