nyoj zb的生日（DFS）

zb的生日

原题链接：http://acm.nyist.net/JudgeOnline/problem.php?pid=325

描述

今天是阴历七月初五，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)分别代表每个西瓜的重量

输出

输出分成两堆后的质量差

样例输入

55 8 13 27 14

样例输出

3

ps：初步接触深搜，自己写的时间复杂度比较高

代码：

#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);    }}

最优代码：

#include <stdio.h>#define max(a,b) a>b?a:bint 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[n]/2;        dfs(n,0);        printf("%d\n",sum[n]-2*ans);    }    return 0;}