dfs组合问题优化

上篇说到了dfs组合，这道题加了个素数判断，但是在核心代码上做了一下优化，做法有点不一样，要体会一下。

问题 C: 【递归入门】组合+判断素数
时间限制: 1 Sec 内存限制: 128 MB
提交: 1545 解决: 441
[提交][状态][讨论版][命题人:外部导入]
题目描述
已知 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

这道题我两天才做出了，查了各种代码，改了无数次，各种细节，还有对dfs的掌握不了解，下面一点一点来讲解，当做完B之后对，对深搜应该有一定的了解了，怎么避免重复遍历，这也是排列和组合不同的地方，B问题在for循环中，初始化的时候用了i=a[x-1]；但明显我们的数据是不能像组合那样重复1，2，3这样，所以我们也还是同一个办法，在初始化的上动手脚，AB中，我都是用了dfs(int x)一个参数，可能是习惯问题，很少尝试，两个或两个以上的参数，但这次用上了，也扩展了自己的知识，下面来附上核心代码

for(int i=temp;i<n;i++){//temp是参数最存储数组的第几位，这样可以控制不再访问低位置的数组 
		if(v[i]!=1){
			v[i]=1;
			p[x]=a[i];
			sum+=p[x];
			dfs(x+1,i+1);
			sum-=p[x];
			v[i]=0;
		}
	}

想想核心代码都很清楚了，看是否遍历过，然后设置访问位，结果加上当前访问的，继续下一次遍历，如果退回当前遍历，有加就有减，恢复访问位。唯一不同的地方就是初始化位置上，i的值，B问题中，我用的是a[x-1]，上面提到，在组合中当然没问题，我访问完1，继续访问2，但是我们的数据不是1,2,3…这么简单，是随机数，不能像B问题中那样，上一步限制这一步，有人说排序，这样可以限制上一步限制下一步了，可以是可以，但是，排序有个问题就是如果数据中有多个相同的数据，这样你在访问位判断的时候还要加一个大小判断，其实我试了，没有成功，可能有哪一步没有想到位吧。在这里我用了一个temp，是上一步传传过来的参数，存储的是访问到数组中的哪个了，这样我就不会想排序这样重复的从1开始，重复遍历，这个其实在dfs（）模板中加多了一个参数，但是操作都是类似的，没多大变化，还是要好好思考一下A,B,C问题。

因为只是在B的基础上改编，加上一个素数判断，没有多大变化，所以就不灵魂画手了。。。

这是题目连接：http://codeup.cn/contest.php?cid=100000608
下面是AC代码

#include<iostream>
#include<string.h>
using namespace std;
int n,k;
int a[101];
int v[101];
int p[101];
long long sum;
long long ans;
bool isprime(long long x){
	if(x==1)
		return false;
	for(int i=2;i*i<=x;i++){
		if(x%i==0)
			return false;
	}
	return true;	
} 

void dfs(int x,int temp){
	if(x==k){
//		for(int i=0;i<k;i++)
//			cout<<p[i]<<" ";
//		cout<<endl; 
		if(isprime(sum)){
//			cout<<sum<<endl;
			ans++;
		}	
		return ;
	}
	for(int i=temp;i<n;i++){//temp是参数最存储数组的第几位，这样可以控制不再访问低位置的数组 
		if(v[i]!=1){
			v[i]=1;
			p[x]=a[i];
			sum+=p[x];
			dfs(x+1,i+1);
			sum-=p[x];
			v[i]=0;
		}
	}
	return ;
}

int main(){
	while(cin>>n>>k){
		memset(a,0,sizeof(a));
		memset(v,0,sizeof(v));
		for(int i=0;i<n;i++){
			cin>>a[i];
		}
		ans=0;
		sum=0;
		dfs(0,0);
		cout<<ans<<endl;
	}
}