自学算法: 【递归入门】组合+判断素数

问题 C: 【递归入门】组合+判断素数

题目描述
已知 n 个整数b1,b2,…,bn
以及一个整数 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）
第二行n个整数：x1,x2，…,xn （1<=xi<=5000000）
————————————————————————————————————————
输出
一个整数（满足条件的方案数）。
————————————————————————————————————————
样例输入 Copy
4 3
3 7 12 19
————————————————————————————————————————
样例输出 Copy
1

**

代码：

**

#include<cstdio>
const int maxn = 1005;
int a[maxn];
int count = 0, n, m;//cout 计算素数数量
bool isPrime(int a)
{
	for(int i = 2; i*i <= a; i++){
		if(a % i == 0)
			return false;
	}
	return true;
}
void DFS(int index,int ans,int number)
{
	if(number == m){
		if(isPrime(ans))
			count++;
		return;
	}
	if(index + 1 <= n){
		DFS(index + 1,ans+a[index],number+1);
		DFS(index + 1,ans,number);
	}
}
int main(void)
{
	scanf("%d %d", &n, &m);
	for(int i = 0; i < n; i++)
		scanf("%d", a+i);
	DFS(0,0,0);
	printf("%d ", count);
}