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

题目链接：http://codeup.cn/problem.php?cid=100000608&pid=2

题目描述
已知 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）

输出
一个整数（满足条件的方案数）。

样例输入
4 3
3 7 12 19

样例输出
1

代码

#include <cstdio> 
#include <cmath>

int b[21], n, k, sum, num;

bool is_prime(int a) {
	if(a == 1)
		return false;
	int q = (int)sqrt(1.0 * a);
	for(int i = 2; i < q; i++) {
		if(a % i == 0)
			return false;
	}
	return true;
}

void DFS(int index, int nowk, int sum) {
	if(nowk == k && is_prime(sum)) {
		num++;
		return ;
	}
	if(index == n || nowk > k)
		return ;
	
	DFS(index + 1, nowk + 1, sum + b[index]);
	DFS(index + 1, nowk, sum);
}

int main() {
	while(scanf("%d%d", &n, &k) != EOF) {
		num = 0;
		for(int i = 0; i < n; i++)
			scanf("%d", &b[i]);
		DFS(0, 0, 0);
		printf("%d\n", num);
	}
	return 0;
}