nyoj325 zb的生日 （深度搜索）（动态规划）

zb的生日

时间限制：3000 ms  |  内存限制：65535 KB

难度：2

描述

今天是阴历七月初五，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

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

此代码借鉴于：http://blog.csdn.net/the_love_story/article/details/51333269

dfs代码：

#include <stdio.h>#include <algorithm>#include <math.h>using namespace std;int w[10010];int tot;int ans;int N;void dfs(int s,int k) //一份的重量和第k+1个西瓜 {if(k == N)return ;ans = min(abs(tot - 2 * s),ans); //一堆西瓜分为tot-s和s两份，tot-2s是两份间的差值 dfs(s + w[k],k + 1); //加w[k]的情况 dfs(s,k + 1);//没加加w[k]的情况 }int main(){while(~scanf("%d",&N)){tot = 0;ans = 99999;for(int i = 0;i < N;i ++ ){scanf("%d",&w[i]);tot += w[i];}dfs(0,0);printf("%d\n",ans);}return 0;}

dp代码：

#include <stdio.h>#include <string.h>#define max(a,b) a > b ? a : bint w[25];int dp[100010];int main(){int N;while(~scanf("%d",&N)){int i,j,sum = 0;memset(dp,0,sizeof(dp));for(i = 0;i < N;i ++ ){scanf("%d",&w[i]);sum += w[i];}int half = sum / 2;for(i = 0;i < N;i ++ )for(j = half;j >= w[i];j -- )dp[j] = max(dp[j],dp[j - w[i]] + w[i]);printf("%d\n",sum - 2 * dp[half]);}return 0;}

优化后的dfs代码：

#include <cstdio>#include <cstring>#define max(a, b) a > b ? a : bint w[22], sum[22];int n, ans, half;void dfs(int t, int s){if(t == 0){//出口 ans = max(ans, s);return ;}if(sum[t] + s <= ans || ans == half)return ;//出口 dfs(t - 1, s);if(s + w[t] <= half)dfs(t - 1, s + w[t]);}int main(){while(~scanf("%d", &n)){int i; ans = -1;for(i = 1;i <= n;i ++){scanf("%d", &w[i]);sum[i] = sum[i - 1] + w[i];}half = sum[n] / 2;dfs(n, 0);printf("%d\n", sum[n] - 2 * ans);}return 0;}