Poj-3977 Subset //折半枚举

题目链接

poj 3977

题意

就是给定 $n$ 个数( $n<=35$ )，然后让你选择一个子集 $S$ ，使得 $S$ 中的数和的绝对值最小，让输出这个最小值和集合 $S$ 大小，如果存在多个值相等的集合，输出集合大小最小的那个。

思路

由于 $n<=35$ 直接枚举子集 $S$ 会超时，所以需要用到一个技巧，就是折半枚举。

折半的话，一般最多 $18$ 个元素，二进制枚举不会超时，将前半段二进制枚举所得各项集合大小和元素值存入 $map$ 中(在此期间也需要维护最优解，因为最优解可能只出现在前半段，后半段也一样)，在后半段二进制枚举时，假设枚举所得结果为 $sum$，就去 $map$ 里去找值靠近 $-sum$ 的值，具体看代码。

参考代码

#include<iostream>
#include<algorithm>
#include<string>
#include<cmath>
#include<map>
using namespace std;
typedef long long ll;
typedef pair<ll,int> P;
const int maxn=40;
#define pb push_back
#define ft first
#define sd second
#define ms(x,y) memset(x,y,sizeof(x))
ll a[maxn];

ll Abs(ll x){ return x>=0?x:-x; }
map<ll,int> m;
int main() {
    int n;
    while(~scanf("%d",&n)){
        if(!n)break;
        m.clear();
        for(int i=0;i<n;i++){
            scanf("%lld",&a[i]);
        }
        P ans(Abs(a[0]),1);//存储最优解
        for(int i=0;i<(1<<(n/2));i++){//先枚举前n/2个
            ll sum=0;
            int num=0;//集合大小
            for(int j=0;j<n/2;j++) {
                if(i&(1<<j)){
                    sum+=a[j];
                    num++;
                }
            }
            if (num==0)continue;
            ans=min(ans,P(Abs(sum),num));//维护前半段的最优解
            if(m.find(sum)==m.end())m[sum]=num;//保证值相同时，集合大小最小
            else m[sum]=min(m[sum],num);
        }
        //在对后n/2个数据进行二进制枚举并维护最优解
        for(int i=0;i<(1<<(n-n/2));i++){
            ll sum=0;
            int num=0;//集合大小
            for(int j=0;j<n-n/2;j++){
                if(i&(1<<j)){
                    sum+=a[n/2+j];
                    num++;
                }
            }
            if(num==0)continue;
            ans=min(ans,P(Abs(sum),num));//维护后半段的最优解
            map<ll,int>::iterator it=m.lower_bound(-sum);//返回大于等于-sum的元素的迭代器
            if(it!=m.end()){//it->first>=-sum
                ans=min(ans,P(Abs(it->ft+sum),it->sd+num));//维护合并后的最优解，集合大小也要合并
            }
            if(it!=m.begin()){//it->first比-sum小一点,也有可能产生一个最小值
                it--;
                ans=min(ans,P(Abs(it->ft+sum),it->sd+num));
            }
        }
        cout<<ans.ft<<' '<<ans.sd<<'\n';
    }
    return 0;
}