dfs的运用
题目描述
已知 nn 个整数 x_1,x_2,…,x_nx1,x2,…,xn ,以及11个整数kk(k<nk<n)。从nn个整数中任选kk个整数相加,可分别得到一系列的和。例如当n=4,k=3n=4,k=3,44个整数分别为3,7,12,193,7,12,19时,可得全部的组合与它们的和为:
3+7+12=223+7+12=22
3+7+19=293+7+19=29
7+12+19=387+12+19=38
3+12+19=343+12+19=34。
现在,要求你计算出和为素数共有多少种。
例如上例,只有一种的和为素数:3+7+19=293+7+19=29。
输入格式
键盘输入,格式为:
n,kn,k(1 \le n \le 20,k<n1≤n≤20,k<n)
x_1,x_2,…,x_n (1 \le x_i \le 5000000)x1,x2,…,xn
(1≤x i≤5000000)
输出格式
屏幕输出,格式为: 11个整数(满足条件的种数)。
输入输出样例
输入 #1
4 3
3 7 12 19
输出 #1
1
#include<stdio.h>
bool isprime(int n)
{
int i;
for(i=2;i*i<=n;i++)
if(n%i==0) return false;
return true;
}
int n,k;
int a[25];
long long ans=0;
void dfs(int m,int sum,int startx)
{
if(m==k)
{
if(isprime(sum))
ans++;
return ;
}
for(int i=startx;i<n;i++)
dfs(m+1,sum+a[i],i+1);
return ;
}
int main()
{
scanf("%d%d",&n,&k);
for(int i=0;i<n;i++)
scanf("%d",&a[i]);
dfs(0,0,0);
printf("%d\n",ans);
return 0;
}