Codeup100000608问题 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）

输出：

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

样例输入：

4 3
3 7 12 19

样例输出：

1

实现代码：

#include <cstdio>
#include <cstring>
#include <math.h>
#include <vector>
using namespace std;

int n;
int r;
int a[25]={0};
vector<int> ans;
int num;

void DFS(int index,int now){
	if(now==r){
		vector<int>::iterator it;
		int sum=0;
		for(it=ans.begin();it!=ans.end();it++){
			sum+=*it;
		}
		if(sum==1){
			return;
		}
		for(int i=2;i<=sqrt(sum);i++){
			if(sum%i==0){
				return;
			}
		}
		num++;
		return;
	}
	if(index==n){
		return;
	}
	ans.push_back(a[index]);
	DFS(index+1,now+1);
	ans.pop_back();
	DFS(index+1,now);
}

int main()
{
	while(scanf("%d%d",&n,&r)!=EOF){
		for(int i=0;i<n;i++){
			scanf("%d",&a[i]);
		}
		num=0;
		DFS(0,0);
		printf("%d\n",num);
		memset(a,0,sizeof(a));
		while(!ans.empty()){
			ans.pop_back();
		}
	}
	return 0;
}