DFS--组合+判断素数

DFS–组合+判断素数

题目描述：

已知 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 <iostream>
#include <cstring>
using namespace std;
#define N 20

int n,k;
int X[N];  //存放输入的n位整数 
int visit[N]; //标志k位加数中有哪些被占
int look[N];//标志n位整数中哪些被占 
int a[N];//标志k位加数分别是哪些
int result=0;

//判断是否为素数 
int fun(int x){
	for(int i=2;i<x;i++){
		if(x%i==0){
			return 1;
		}
	}
	return 0;
}

void dfs(int step){
	//当k位加数已经放满时，达到边界条件
	if(step==k){
		int sum=0;
		//求和
		for(int i=0;i<k;i++){
			cout<<a[i]<<" ";
			sum+=a[i];	
		}
		cout<<sum;
		cout<<endl;
		//判断和是否为素数
		if(!fun(sum)){	
			result++;
		}
		return;
	}
	int j;
	for(int i=0;i<n;i++){
		for(j=0;j<k;j++){
			if(visit[j]==0){
				break;
			}
		}
		if(look[i]==0&&X[i]>a[j-1]){ //控制顺序，以免重复 
			a[j]=X[i];
			look[i]=1; 
			visit[j]=1;
			dfs(step+1);
			visit[j]=0;//回溯
			look[i]=0;//回溯
		}
	}
}

int main(){
	cin>>n>>k;
	//初始化
	memset(X,-1,sizeof(X));
	memset(visit,0,sizeof(visit));
	memset(look,0,sizeof(look));
	memset(a,0,sizeof(a));
	for(int i=0;i<n;i++){
		cin>>X[i];
	}
	dfs(0);
	cout<<result;
}

---

样例结果截图：

1-2行：输入；
3-6行：全部组合
7行：最终和为素数的组合个数

---

因为不能测试，所以也不知道超时了没有~
如有不对，请各位大佬指教！嘻嘻嘻~