您的位置:首页 > 其它

nyoj zb的生日(DFS)

2016-12-08 16:05 274 查看
[align=center]zb的生日[/align]
[align=left]原题链接:http://acm.nyist.net/JudgeOnline/problem.php?pid=325[/align]
描述今天是阴历七月初五,acm队员zb的生日。zb正在和C小加、never在武汉集训。他想给这两位兄弟买点什么庆祝生日,经过调查,zb发现C小加和never都很喜欢吃西瓜,而且一吃就是一堆的那种,zb立刻下定决心买了一堆西瓜。当他准备把西瓜送给C小加和never的时候,遇到了一个难题,never和C小加不在一块住,只能把西瓜分成两堆给他们,为了对每个人都公平,他想让两堆的重量之差最小。每个西瓜的重量已知,你能帮帮他么?

输入多组测试数据(<=1500)。数据以EOF结尾

第一行输入西瓜数量N (1 ≤ N ≤ 20)

第二行有N个数,W1, …, Wn (1 ≤ Wi ≤ 10000)分别代表每个西瓜的重量

输出输出分成两堆后的质量差
样例输入
5
5 8 13 27 14


样例输出
3


[align=left]ps:初步接触深搜,自己写的时间复杂度比较高[/align]
[align=left]代码:[/align]
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
int n;
int a[22],s;
void dfs(int s1,int s2,int i)
{
if(i==n)
{
if(abs(s1-s2)<s)
s=abs(s1-s2);
return ;
}
dfs(s1+a[i+1],s2,i+1);
dfs(s1,s2+a[i+1],i+1);
}
int main()
{
while(~scanf("%d",&n))
{
memset(a,0,sizeof(a));
int i;
for(i=1;i<=n;i++)
{
scanf("%d",&a[i]);
}
s=0xffffff;
dfs(0,0,0);
printf("%d\n",s);
}
}

[align=left]最优代码:[/align]
#include <stdio.h>
#define max(a,b) a>b?a:b
int V,ans,n,w[21],sum[21];
void dfs(int i,int cnt)
{
if(i == 0)//到达底部时
{
ans = max(ans,cnt);//找到最接近V的
return ;
}
if(ans == V || cnt+sum[i] <= ans)//已经找到v结束或者cnt加上前面的所有元素之和比ans小,说明此路不优,结束
return ;
if(cnt+w[i] <= V)
dfs(i-1,cnt+w[i]);
dfs(i-1,cnt);
}
int main()
{
while(~scanf("%d",&n))
{
ans = 0;
for(int i=1; i<=n; i++)
{
scanf("%d",&w[i]);
sum[i] = sum[i-1] + w[i];
}
V = sum
/2;
dfs(n,0);
printf("%d\n",sum
-2*ans);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: