题目描述 Description
已知 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)。
输入描述 Input Description
键盘输入,格式为:
n , k (1<=n<=20,k<n)
x1,x2,…,xn (1<=xi<=5000000)
输出描述 Output Description
屏幕输出,格式为:
一个整数(满足条件的种数)。
样例输入 Sample Input
4 3
3 7 12 19
样例输出 Sample Output
1
数据范围及提示 Data Size & Hint
(1<=n<=20,k<n)
(1<=xi<=5000000)
#include <cstdio>
#include <iostream>
#include <cstring>
#include <cmath>
using namespace std;
int a[5000],n,k,ans = 0;
bool pd(int k)
{
if(k == 1)
return false;
if(k == 2)
return true;
for(int i = 2; i <= sqrt(k) ; i ++)
{
if(k%i==0)
return false;
}
return true;
}
void dfs(int cur,int cnt,int sum)
{
if(cnt == k)
{
if(pd(sum))
{
ans++;
return ;
}
}
for(int i = cur; i <=n; i++)
{
dfs(i+1,cnt+1,sum+a[i]);
}
}
int main()
{
scanf("%d%d",&n,&k);
for(int i = 1; i <= n ; i ++)
{
scanf("%d",&a[i]);
}
dfs(1,0,0);
cout <<ans<<endl;
return 0;
}