题目描述 https://www.luogu.org/problemnew/show/P1120
将木棍排序,从大到小枚举,再加各种剪枝。
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
int a[70],fl[70],next[70],n,m,minn,sum,f,len,cnt;
bool cmp(const int &a,const int &b)
{
return a>b;
}
void dfs(int now,int last,int rest)//now表示第now跟木棍
{ //last是第now跟木棍的上一个编号 rest表示第now跟还需要的长度
int i;
if(!rest)//这根拼好 可以拼下一根
{
if(now==m){f=1; return;}//可行 返回输出
for(i=1;i<=cnt;i++) if(!fl[i]) break;
fl[i]=1;
dfs(now+1,i,len-a[i]);
fl[i]=0;
if(f) return;
}
for(int i=last+1;i<=cnt;i++)
if(!fl[i]&&a[i]<=rest)
{
fl[i]=1;
dfs(now,i,rest-a[i]);
fl[i]=0;
if(f) return;
if(rest==a[i]||rest==len) return;//剪枝
i=next[i];//这种情况不可行 后面与它相等的木棍都不可行
if(i==n) return;
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
int z;
scanf("%d",&z);
if(z>50) continue;
a[++cnt]=z;
minn=max(minn,z);
sum+=z;
}
sort(a+1,a+cnt+1,cmp);
next[cnt]=cnt;
for(int i=cnt-1;i>0;i--)
{
if(a[i]==a[i+1]) next[i]=next[i+1];
else next[i]=i;
}//预处理出相等的木棍
for(int i=minn;i<=sum;i++)
{
if(sum%i!=0) continue;
memset(fl,0,sizeof(fl));
len=i; m=sum/i; f=0;
fl[1]=1;
dfs(1,1,len-a[1]);
if(f==1){ printf("%d",len); break;}
}
return 0;
}