选数
已知 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<iomanip>
using namespace std;
int n,k,a[30],x[30],s=0
int prime(int t)//判断素数
{
int i;
for(i=2;i*i<=t;i++)
{
if(t%i==0) return 0;
}
return 1;
}
void dfs(int w)//层数
{
int i;
if(w>k)
{
int sum=0;
for(i=1;i<=k;i++)
{
sum+=x[a[i]];//对所有值求和
}
if(prime(sum))//判断是否是素数
{
s++;
}
return;
}
for(i=a[w-1]+1;i<=n;i++)
{
a[w]=i;//当前w选择了x个数中第i个位置的数字
dfs(w+1);
}
}
int main()
{
int i;
cin>>n>>k;
for(i=1;i<=n;i++)
cin>>x[i];//存数
dfs(1);
cout<<s;
return 0;
}