Wishare D题

题目：教练评估了每一个人的能力值，现在想要将队伍分成两组进行对抗。一个队的能力值为该队所有人能力值之和。教练希望两队的能力值尽可能接近，但是不知道如何分组。教练希望你能够帮助他解决这个问题。

输入：

一个整数n，表示有n个人。 2 <= n <= 32

a1，a2...an表示每个人能力值。1 <= ai <= 10^12

输出：

当两队能力值最接近时，两队能力值差的绝对值。

 

思维题 。（数据范围太大，不能用dp[sum / 2]的思路）

思路见注释。通过状态压缩求各种组合的值（此处为差值），思路巧妙。

#include <bits/stdc++.h>
using namespace std;
#define ll long long
vector<ll> v1,v2;
ll a[40];
int n;

int main()
{
    cin>>n;
    for (int i=1;i<=n;i++)
    {
        cin>>a[i];
    }
    for (int i=0;i<(1<<(n/2));i++)        状压，某位为0时加上该位对应下标的数，为1时减去，通过这样两层循环求出前一半的数的所有差的组合。如1 2 3 4四个数，前一半为1 2,它们的差值组合为(1 + 2) - 0，1 - 2, 2 - 1, 0 - (1 + 2)
    {
        ll x=0;
        for (int j=0;j<n/2;j++)        
        {
            if ((1<<j)&i) x-=a[1+j];
            else x+=a[1+j];
        }
        v1.push_back(x);
    }
    for (int i=0;i <(1<<(n-n/2));i++)        //求出后一半差值组合
    {
        ll x=0;
        for (int j=0;j<n-n/2;j++)    
        {
            if ((1<<j)&i) x+=a[n/2+1+j];
            else x-=a[n/2+1+j];
        }
        v2.push_back(x);
    }
    ll ans=0x3f3f3f3f3f3f3f3fLL;
    int p=-1;
    sort(v1.begin(),v1.end());
    sort(v2.begin(),v2.end());
    for (vector<ll>::iterator it=v1.begin();it!=v1.end();it++)    //遍历v1，在后一半中找出一个数与当前数值加和最接近0的，每次更新ans即求解
    {
    	ll x=*it;
        int k=lower_bound(v2.begin(),v2.end(),-x)-v2.begin();
        if (k<v2.size())
        {
        	ans=min(ans,abs(v2[k]+x));
        }
        if (k>0)
        {
        	ans=min(ans,abs(v2[k-1]+x));
        }
    }
    cout<<ans<<endl;
    return 0;
}