nyoj 325 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)分别代表每个西瓜的重量

输出

输出分成两堆后的质量差

样例输入

5

5 8 13 27 14

样例输出

3

来源

ural

上传者

李文鑫

 

分析：每一个西瓜有选与不选两种状态，考虑到N <= 20,

因此，可以尝试用简单的搜索解决，时间复杂度为O(2^N)

/*
普通深搜：
  时间：O(2^n)
*/
#include<iostream>
#include<cmath>
using namespace std;
const int maxn = 20 + 10;
int mallon[maxn];
int n, weight, minimum;
void dfs(int index, int sum) {
  if (index == n) {
    if (abs(weight - 2 * sum) <  minimum) minimum = abs(weight - 2 * sum);
    return;
  }
  dfs(index + 1, sum + mallon[index]);
  dfs(index + 1, sum);
}
int main() {
  while (cin >> n) {
    weight = 0;
    for (int i = 0; i < n; i++) {
      cin >> mallon[i];
      weight += mallon[i];
    }
    minimum = weight;
    dfs(0, 0);
    cout << minimum << endl;
  }
  return 0;
}

通过观察上图，不难发现，其实sum一旦超过weight/2时就没有必要往下搜索了。因此可以改进一下：

void dfs(int index, int sum) {
  if (sum >= weight / 2 && index != n) return;
  if (index == n) {
    if (abs(weight - 2 * sum) <  minimum) minimum = abs(weight - 2 * sum);
    return;
  }
  dfs(index + 1, sum + mallon[index]);
  dfs(index + 1, sum);
}