P1120 小木棍 [数据加强版]
题目描述
乔治有一些同样长的小木棍,他把这些木棍随意砍成几段,直到每段的长都不超过5050。
现在,他想把小木棍拼接成原来的样子,但是却忘记了自己开始时有多少根木棍和它们的长度。
给出每段小木棍的长度,编程帮他找出原始木棍的最小可能长度。
输入输出格式
输入格式:
共二行。
第一行为一个单独的整数N表示砍过以后的小木棍的总数,其中N≤65N≤65
(管理员注:要把超过50的长度自觉过滤掉,坑了很多人了!)
第二行为N个用空个隔开的正整数,表示N根小木棍的长度。
输出格式:
一个数,表示要求的原始木棍的最小可能长度
输入输出样例
输入样例#1:
9
5 2 1 5 2 1 5 2 1
输出样例#1:
6
#include <iostream>
#include <algorithm>
#include <functional>
#include <cstdio>
#include <cmath>
using namespace std;
int n;
int num=0;//多少个有效长度
int stick[70];
int next_len[70];
int maxx=-1;
int sum=0;//有效长度之和
bool ok;
bool used[70];
void scanff()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
int temp;
scanf("%d",&temp);
if(temp>50) continue;
stick[++num]=temp;
if(temp<minn) minn=temp;
if(temp>maxx) maxx=temp;
sum+=temp;
}
}
void pre_()
{
for(int i=2;i<=num;i++)
if(stick[i]!=stick[i-1])
for(int j=i-1;!next_len[j]&&j>0;j--)
next_len[j]=i;
}
//len是尝试拼成短木棍长度,all_num是长棍数目,now是现在在拼哪个棍子
//rest是当前棍子还差多少,last是当前棍子上一个用到的短棍
//每次深搜完都要检测是否ok,若ok就直接return
//保证有used[i]就有!used[i],避免memset浪费时间
void dfs(int len,int all_num,int now,int rest,int last)
{
//如果rest=0,当前棍子拼好,开始拼下一个棍子
//先放上去当前未用短最长的棍子,因为它总要被放
if(!rest)
{
//剪枝4:拼完最后一根,直接返回ok
if(now==all_num)
{
ok=true;
return;
}
//贪心思想:最大未用短棍子总要被放,不如先放
for(int i=1;i<=n;i++)
if(!used[i])
{
used[i]=true;
dfs(len,all_num,now+1,len-stick[i],i);
used[i]=false;
if(ok) return;
break;
}
}
//当前长棍没被装完
else
{
//剪枝5:二分法找到上一个棍子之后第一个能放的
//因为放上一个之前的一定已被深搜过
int l=last+1,r=num,mid;
while(l<r)
{
mid=(l+r)>>1;
if(stick[mid]<=rest)
r=mid;
else l=mid+1;
}
for(int i=l;i<=num;i++)
{
//剪枝6:用过的直接跳过
if(used[i]) continue;
used[i]=true;
dfs(len,all_num,now,rest-stick[i],i);
used[i]=false;
if(ok) return;
//剪枝7:搜索失败,跳过和当前棍长度相同的短棍
//如果所有剩余棍都和当前长度相同(最短),搜索结束
if(next_len[i]==0) break;
//重要剪枝8:如果当前棍装满此棍子但搜索失败,如果有其他方案:
//(若干个其他小棍填满此长棍),可交换当前棍与这些小棍依然成立
//可知有矛盾,无需继续搜索
if(rest==stick[i]) break;
i=next_len[i]-1;
}
}
}
void work()
{
sort(stick+1,stick+num+1,greater<int>());//先将其从大到小排列
pre_();//记录比每一个棍短短下一个棍短位置;最小的棍记为0
//剪枝1:长棍长度一定大于最长的短棍长度;
//剪枝2:如果长棍大于 总长度/2,那么只能为总长度
for(int i=maxx;i<=sum/2;i++)
{
if(sum%i!=0)//剪枝3:不能整除总长度,肯定不合理
continue;
used[1]=true;//最大的总要放,所以放在第一个
dfs(i,sum/i,1,i-stick[1],1);
used[1]=false;
if(ok)
{
printf("%d\n",i);
return;
}
}
printf("%d\n",sum);
}
int main()
{
freopen("in.in","r",stdin);
freopen("out.out","w",stdout);
scanff();
work();
return 0;
}