AcWing 167. 木棒 dfs+剪枝

Posted 2021-11-07 karshey

题
参考1
参考2

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
#define pb push_back
#define fi first
#define se second
#define mem(a,x) memset(a,x,sizeof(a));
#define db double 
#define fir(i,a,n) for(int i=a;i<=n;i++)
#define debug(x) cout<<#x<<" "<<x<<endl;
const int inf=0x3f3f3f3f;
const int MOD=1e9+7;
//======================
const int N=70;
int n;
int a[N];
int val,sum;
int target=0,cnt=0;//目标长度 目标个数 
int v[N];
bool cmp(int a,int b)
{
	return a>b;
}
bool dfs(int done,int lennow,int now)//已经完成了几个 当前单个木条的长度是几 小木块的序号
{
	if(done>=cnt) return 1;
	if(lennow==target) return dfs(done+1,0,1);
	
	int fail=0;
	for(int i=now;i<=n;i++)
	{
		if(!v[i]&&lennow+a[i]<=target&&a[i]!=fail) //没用过 不会超出范围 没有黑历史
		{
			v[i]=1;
			if(dfs(done,lennow+a[i],i+1)) return 1;
			v[i]=0;
			fail=a[i];//走到这里说明不行
			
			//走到这里 一个没有 说明i=now的不行（这样分配的方案完全不可能）
			//lennow+a[i]==target 都等于了还不行 说明后面的太拉了 不能满足这种方案 
			if(lennow==0||lennow+a[i]==target) return 0;			 
		} 
	}
	return 0;
} 
int main()
{
	while(cin>>n&&n)
	{
		sum=0;
		val=-0x3f3f3f3f;
		for(int i=1;i<=n;i++) 
		{
			cin>>a[i];
			sum+=a[i];
			val=max(val,a[i]);
		}
		sort(a+1,a+1+n,cmp);
		
		for(int i=val;i<=sum;i++)
		{
			if(sum%i==0)
			{
				target=i;
				cnt=sum/target;
				mem(v,0);
				if(dfs(0,0,1)) break;
			}		
		}
		
		cout<<target<<endl;
	}
		
	return 0; 
}