题目
https://www.luogu.org/problemnew/show/P1120
思路
预先处理出所有木棍的总长度,且保证枚举答案的值能被总长度整除。
每根木棍的长度可用桶来存储,并且预先处理出最长的和最短的木棍的长度,搜索时从最大长度到最小长度递减枚举。
若拼接当前木棍时已用了一根长为X的木棍,则dfs时从长度X开始搜索。
若某组拼接不成立,且此时 已拼接的长度为0 或 当前已拼接的长度与刚才枚举的长度之和为最终枚举的答案 时,则可直接跳出循环,因为此时继续枚举其它更小的值时,显然可能情况更少,且同样凑不完。
代码
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=70;
int n,a[N],cnt,ans,s;
bool cmp(int x,int y)
{
return x>y;
}
bool b[N];
inline void dfs(int ass,int sum,int g,int p)
{
if(sum*g==ans)
{
printf("%d\n",g);
exit(0);
}
if(ans-ass<a[cnt])return;
if(ass==g)
{
dfs(0,sum+1,g,1);
return;
}
for(int i=p;i<=cnt;i++)
if(!b[i]&&ass+a[i]<=g)
{
b[i]=1;
dfs(ass+a[i],sum,g,i+1);
b[i]=0;
if(ass+a[i]==g||ass==0)break;
while(a[i]==a[i+1])i++;
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
int x;
scanf("%d",&x);
if(x<=50){a[++cnt]=x; ans+=a[cnt]; s=max(a[cnt],s);}
}
sort(a+1,a+cnt+1,cmp);
for(int i=s;i<=ans/2;i++)
if(ans%i==0) dfs(0,0,i,1);
printf("%d\n",ans);
}