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<<length<<endl;
		break;
	}
	length++;
}

第二，可行性剪枝

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

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

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

第三，冗余性剪枝

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

int j=i;
while(j<n&&w[j]==w[i]) j++;
i=j-1;

完整代码（注意当一组完成时，下一组从0开始搜）

#include<iostream>
#include<algorithm>
#include<cstring>
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;i<n;i++)
	{
		if(vis[i]) continue;
		if(s+w[i]>length) 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&&w[j]==w[i]) j++;
		i=j-1;
	}
	return false;
}
int main()
{
	while(cin>>n&&n)
	{
		memset(vis,0,sizeof vis);
		sum=0;
		for(int i=0;i<n;i++)
		{
			cin>>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<<length<<endl;
				break;
			}
			length++;
		}
	}
}