礼物分配（折半搜索）

D - 礼物分配

题目描述

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

输入

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

输出

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

解题方法

折半搜索，将礼物分成两半，如果是奇数，前一半要比后一半多一个（原因在代码中解释），前一半暴力，二进制枚举各种取法，1代表Eric拿，0表示R.W拿，然后cnt是Eric拿的数目，将cnt相同的sum1-sum2 存入一个vector[cnt]，然后同样遍历后一半，二分查找最接近-x的值（二分的细节在代码中解释）

#include <iostream>
#include <algorithm>
#include <vector>

using namespace std;

const int INF=0x3f3f3f3f;
int w[35],v[35],n;
vector <int> q[35];

int binary_find(int n,int x)
{
    int l=0,r=q[n].size()-1,res=-1;  //res是为标记二分结果是否在端点处
    while(l<r){
        int mid= (l+r)/2;
        if(q[n][mid] >= -x){
            r=mid;
            res=mid;
        }
        else l=mid+1;
    }
    if(res == -1) return abs(q[n][r]+x);  
    if(res == 0) return abs(q[n][0]+x);
    return min(abs(q[n][l]+x),abs(q[n][l-1]+x));
}

int main()
{
    int t;
    cin >> t;
    while(t--){
        int n; cin >> n;
        int ans = INF;
        for(int i=0;i<n;i++) cin >> v[i];
        for(int i=0;i<n;i++) cin >> w[i];

        int m2=n/2,m1=n-m2;
        for(int i=0;i<=m1;i++) q[i].clear();
        for(int i=0;i< 1<<m1;i++){
            int sum1=0,sum2=0,cnt=0;
            for(int j=0;j<m1;j++){
                if(1<<j & i){
                    sum1 += v[j];
                    cnt++;
                }

                else sum2 += w[j];
            }
            q[cnt].push_back(sum1-sum2);
        }

        for(int i=0;i<=m1;i++)
            sort(q[i].begin(),q[i].end());

        for(int i=0;i< 1<<m2;i++){
            int sum1 = 0,sum2 = 0,cnt = 0;
            for(int j=m1;j<n;j++){
                if(1<<(j-m1) & i){
                    sum1 += v[j];
                    cnt++;
                }
                else sum2 += w[j];
            }

            //n若为奇数  需要二分两次
            if(n&1) ans = min(ans , binary_find(m2-cnt,sum1-sum2));   //若m2比m1大，那么m2-cnt且（cnt）是0的时候 会大于m1  vector内元素为空

            ans = min(ans,binary_find(m1-cnt,sum1-sum2));

        }
        cout << ans << endl;
    }

    return 0;
}