NOI-1819 木棒(搜索dfs)

题目传送门

描述

乔治拿来一组等长的木棒,将它们随机地裁断,使得每一节木棍的长度都不超过50个长度单位。然后他又想把这些木棍恢复到为裁截前的状态,但忘记了初始时有多少木棒以及木棒的初始长度。请你设计一个程序,帮助乔治计算木棒的可能最小长度。每一节木棍的长度都用大于零的整数表示。
输入
输入包含多组数据,每组数据包括两行。第一行是一个不超过64的整数,表示砍断之后共有多少节木棍。第二行是截断以后,所得到的各节木棍的长度。在最后一组数据之后,是一个零。
输出
为每组数据,分别输出原始木棒的可能最小长度,每组数据占一行。
样例输入

9
5 2 1 5 2 1 5 2 1
4
1 2 3 4
0

样例输出

6
5

代码

#include<bits/stdc++.h>
using namespace std;
int n,sum,num;
bool flag;
int a[70],vis[70];
void dfs(int x,int pid,int cnt)
{
	if(flag) return;
	if(x==pid) cnt++,x=0;
	if(cnt==sum/pid)
	{
		flag=true;
		return;
	} 
	for(int k=n-1;k>-1;k--)
	{
		if(x+a[k]>pid||vis[k]) continue;
		if(a[k]==a[k-1]&&!vis[k-1]) continue;
		vis[k]=1;
		dfs(x+a[k],pid,cnt);
		vis[k]=0;
		if(flag) return;
		if(x==0) return;
	}
}
int main()
{
	while(cin>>n)
	{
		if(n==0) break;
		memset(a,0,sizeof(a));
		sum=0;num=0;
		for(int i=1;i<=n;i++)
		{
			int tmp;
			scanf("%d",&tmp);
			if(tmp<=50)
			{
				a[num++]=tmp;
				sum+=tmp;
			}
		}
		sort(a,a+n);
		for(int i=a[n-1];i<=sum;i++)
		{
			if(sum%i!=0) continue;
			flag=false;
			dfs(0,i,1);
			if(flag)
			{
				cout<<i<<endl;
				break;
			}
		}
	}
	return 0;
}

一道搜索题目,重点在于如何利用深度搜索去配对木棒。其中,为了减少重复操作,对木棒长度进行排序,从最长的木棒开始进行搜索,越短的木棒灵活性越大,若所有木棒匹配成功,则返回true。
PS:一开始写的dfs返回值不能保证当匹配不成功时一定返回false,导致wa;后来由于是从短木棒开始搜索而导致超时。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值