Acwing 167. 木棒(DFS顺序优化+剪枝)

167. 木棒 - AcWing题库

题意是让把所有的木棒分成长度相同的几组,长度最小

首先应该先从小到大枚举长度吧,然后将可以整除的长度DFS判断一下

可以想到直接DFS情况会很多,超时警告!想办法进行优化


第一,根据搜索顺序优化,枚举长度小的分支比长度大的分支要多,所以先枚举长度大的

sort(w,w+n);
reverse(w,w+n);

然后对可以整除的DFS

//第u组,当前长度s,该组枚举起点 
bool dfs(int u,int s,int start);
length=1;
while(1)
{
	if(sum%length==0&&dfs(0,0,0))
	{
		cout<

第二,可行性剪枝

如果在第u组放置失败,且此时第u组长度为0,说明存在有的木棒长度大于我们设定的长度,所以这个长度一定不可行,直接false

另外如果放置失败,且如果成功则正好为length时,由于往后的木棒长度<=w[i],所以往后该组必然放置不成功,直接false

if(!s) return false;
if(s+w[i]==length) return false;

第三,冗余性剪枝

如果当前未放置成功,则后面和该木棒长度相等的也一样,直接略过即可

int j=i;
while(j

完整代码(注意当一组完成时,下一组从0开始搜)

#include
#include
#include
using namespace std;
const int N=70;
int n;
int w[N],sum,length;
bool vis[N];
//第u组,第u组当前长度,第u组的枚举位置 
bool dfs(int u,int s,int start)
{
    //开了u组,每组都是length
	if(u*length==sum) return true;
	//开新组 
	if(s==length) return dfs(u+1,0,0);
	//在同一组中,编号从小到大排列(经过有序排序了) 
	for(int i=start;ilength) continue;
		vis[i]=true;
		if(dfs(u,s+w[i],i+1)) return true;
		vis[i]=false;
		//第一根放置失败 
		if(!s) return false;
		//最后一根放置失败 
		if(s+w[i]==length) return false;
		//第i根失败,其后相同长度的也失败 
		int j=i;
		while(j>n&&n)
	{
		memset(vis,0,sizeof vis);
		sum=0;
		for(int i=0;i>w[i];
			sum+=w[i];
		}
		sort(w,w+n);
		reverse(w,w+n);
		length=1;
		while(1)
		{
			if(sum%length==0&&dfs(0,0,0))
			{
				cout<

你可能感兴趣的:(深度优先,剪枝,算法)