问题 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 <iostream>
#include <bits/stdc++.h>
using namespace std;
int isprime(int x){
	int sqr = (int)sqrt(1.0 * x);
	for(int i = 2; i <= sqr; i++){
		if(x % i == 0){
			return 0;
		}
	}
	return 1;
}
int cnt = 0, n, k;
void DFS(int a[], int index, int nowk, int sum){
	if(nowk == k){
		if(isprime(sum)){
			cnt++;
		}
		return;
	}
	if(index == n) return;
	DFS(a, index + 1, nowk + 1, sum + a[index]);//选index这个位置的数
	DFS(a, index + 1, nowk, sum);//不选index这个位置的数
}
int main(int argc, char** argv) {
	while(cin >> n >> k){
		int a[n];
		cnt = 0;
		for(int i = 0; i < n; i++){
			cin >> a[i];
		}
		DFS(a, 0, 0, 0);
		cout << cnt << endl;
	}	
	return 0;
}