FZUOJ Problem 2178 礼物分配

                                                                                      Problem 2178 礼物分配

题目链接： Click Here~ 

Problem Description

在双胞胎兄弟Eric与R.W的生日会上，他们共收到了N个礼物，生日过后他们决定分配这N个礼物(numv+numw=N)。对于每个礼物他们俩有着各自心中的价值vi和wi，他们要求各自分到的礼物数目|numv-numw|<=1，并且各自所衡量的礼物价值的差值|sumv-sumw|尽可能小，现在他们想知道最小的差值是多少。

 Input

第一行为一个整数表示数据组数T。 接下来T组数组，每组数据第一行为一个整数N。(N<=30) 第二行有N个整数，表示Eric所衡量的每个礼物的价值vi。(1<=vi<=10000000) 第三行也有N个整数，表示R.W所衡量的每个礼物的价值wi。(1<=wi<=10000000)

 Output

对于每组数据，输出最小的差值。

算法分析：

可以容易想到的方法是直接枚举暴力。现在对第一个人来分析，每个物品都有选或者不选的可能。这样的话总情况是2^30次方。肯定超时。想着怎么优化。很容易的想到了二分搜索。就是折半查找。思想就好是先预处理出前一半部分的结果。然后，在用前面的结果判断后面的情况。这样就可以优化到了2^15次方了。

#include <iostream>
#include <algorithm>
#include <vector>
#include <cstdio>
#include <cstring>
using namespace std;

typedef __int64 LL;
const int INF = 1 << 30;
const int MAXN = 40;
vector<int> num[MAXN];
int vi[MAXN],wi[MAXN];

int main() {
   int n,T;
   scanf("%d",&T);
   while(T--) {
       scanf("%d",&n);
       for(int i = 0;i < n;++i) {
           scanf("%d",&vi[i]);
       }
       for(int i = 0;i < n;++i) {
           scanf("%d",&wi[i]);
       }

       for(int i = 0;i <= n;++i)
         num[i].clear();

       int n2 = n/2;
       int cnt,sum1 ,sum2,sum;
       for(int S = 0;S < 1 << n2; ++S) {
          cnt = 0,sum1 = 0,sum2 = 0;
          for(int i = 0;i < n2;++i) {
              if(S >> i & 1) {
                  sum1 += vi[i];
                  cnt++;
              } else {
                  sum2 += wi[i];
              }
          }
          num[cnt].push_back(sum1 - sum2);
       }

       for(int i = 0;i < n2;++i) {
           sort(num[i].begin(),num[i].end());
           num[i].erase(unique(num[i].begin(),num[i].end()),num[i].end());
       }

       int ans = INF;
       for(int S = 0;S < 1 << (n-n2);++S) {
           sum,cnt = 0,sum1 = 0,sum2 = 0;
           for(int i = 0;i < (n-n2);++i) {
               if(S >> i & 1) {
                   sum1 += vi[i+n2];
                   cnt++;
               } else {
                   sum2 += wi[i+n2];
               }
           }
           int t = n - n2 - cnt;
           sum = sum1 - sum2;    
           vector<int>::iterator iter;
           iter = lower_bound(num[t].begin(),num[t].end(),-sum);

           if(iter != num[t].end() && abs(*iter + sum) < ans)
               ans = abs(*iter + sum);

           if(iter != num[t].begin()) {
                --iter;
               if(abs(*iter + sum) < ans) ans = abs(*iter + sum);
           }
       }

        printf("%d\n",ans);
   }
   return 0;
}


/*


3
1 2 3
4 2 1

5
1 2 3 5 4
1 1 1 1 5

6
1 2 3 4 5 5
1 1 1 1 1 8

*/