学习笔记（DFS加回溯）洛谷1036

上次，我用暴力枚举。这次，我们用DFS回溯的方式来解题。

这一题，是洛谷的深搜第一题，我觉得还是有点难。难度在于，如何在n个数中选取k个，且不重复。因此，我第一次的代码是这样的。我是想每一次选数，都从这一次选择的数的下一个位置开始。上代码：

#include<cstdio>
using namespace std;
int a[25];
int b[25];
int v[25];
int ans;
int n,k;
bool isPrime(int n)
{
	for(int i=2;i*i<=n;++i)
	{
		if(n%i==0) return false;
	}
	return true;
}

void dfs(int m,int sum)
{
	if(m==k+1)
	{
		if(isPrime(sum)) ans++;
		return;	
   }

	for(int i=m;i<=n;++i)
	{
		if(v[i]==0)
		{
			b[m]=a[i];
			v[i]=1;
			sum+=b[m];
			dfs(m+1,sum);
			v[i]=0;
			sum-=b[m];			
		}
	}
}
int main()
{	
	freopen("P1036_1.in","r",stdin);
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;++i) scanf("%d",&a[i]);
	dfs(1,0);
	printf("%d\n",ans);
	return 0;	
}

可以看到，for循环int i=m;i<=n;++i   我是计划，用m（层数）来控制选数的位置。因为层数是递进的。但是由于有回溯，我经过调试，发现如果你到了第三层，回溯下来，有可能v[3]=1,m的值却变小了，m=2,出现过这种情况。有可先找第三个数，再找第2个数，形成重复选择。

我原来是因为看到题解的写法，都是传三个参数，我就想只传两个参数试一下。发现，这个位置还必须要单独传一个参数来确定，因此，我只能多传一个参数pos,它的值就是这回选数的下一位置，（i+1），上代码：

#include<cstdio>
using namespace std;

int a[25];
int b[25];
int v[25];
int ans;
int n,k;
bool isPrime(int n)
{
	for(int i=2;i*i<=n;++i)
	{
		if(n%i==0) return false;
	}
	return true;
}

void dfs(int m,int pos,int sum)
{
	if(m==k+1)
	{
		if(isPrime(sum)) ans++;
		return; 	    
    }

	for(int i=pos;i<=n;++i)
	{
		if(v[i]==0)
		{
			b[m]=a[i];
			v[i]=1;
			sum+=b[m];
			dfs(m+1,i+1,sum);
			v[i]=0;
			sum-=b[m];			
		}
	}
}
int main()
{	
	freopen("P1036_1.in","r",stdin);
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;++i) scanf("%d",&a[i]);
	dfs(1,1,0);
	printf("%d\n",ans);
	return 0;	
}