Subset---poj3977(折半枚举+二分查找)

题目链接:http://poj.org/problem?id=3977

给你n个数,找到一个子集,使得这个子集的和的绝对值是最小的,如果有多种情况,输出子集个数最少的;

n<=35,|a[i]|<=10e15

子集个数共有2^n个,所以不能全部枚举,但是可以分为两部分枚举;

枚举一半的所有情况,然后后一半二分即可;

#include<iostream>
#include<algorithm>
#include<string.h>
#include<stdio.h>
#include<math.h>
#include<vector>
#include<map>
using namespace std;
#define N 45
#define PI 4*atan(1.0)
#define mod 1000000007
#define met(a, b) memset(a, b, sizeof(a))
#define INF 10000000000000000
typedef long long LL;

int n;
LL a[N];

LL Abs(LL x)
{
    return x<0?-x:x;
}

int main()
{
    while(scanf("%d", &n), n)
    {
        for(int i=0; i<n; i++)
            scanf("%I64d", &a[i]);

        map<LL, int> M;
        map<LL, int>::iterator it;
        pair<LL, int> ans(Abs(a[0]), 1);

        for(int i=1; i<1<<(n/2); i++)
        {
            LL sum = 0;int cnt = 0;
            for(int j=0; j<(n/2); j++)
            {
                if((i>>j)&1)
                {
                    sum += a[j];
                    cnt ++;
                }
            }
            ans = min(ans, make_pair(Abs(sum), cnt));///全部是前半部分的;
            if(M[sum])///更新cnt为小的;
                M[sum] = min(M[sum], cnt);
            else
                M[sum] = cnt;
        }

        for(int i=1; i<1<<(n-n/2); i++)
        {
            LL sum = 0;int cnt = 0;
            for(int j=0; j<(n-n/2); j++)
            {
                if((i>>j)&1)
                {
                    sum += a[j+n/2];
                    cnt ++;
                }
            }
            ans = min(ans, make_pair(Abs(sum), cnt));///全部是后半部分的;

            it = M.lower_bound(-sum);///找到第一个大于-sum的位置,然后取两种情况的最小值;

            if(it != M.end())
                ans = min(ans, make_pair(Abs(sum+it->first), cnt+it->second));
            if(it != M.begin())
            {
                it--;
                ans = min(ans, make_pair(Abs(sum+it->first), cnt+it->second));
            }
        }
        printf("%I64d %d\n", ans.first, ans.second);
    }
    return 0;
}
View Code

 

posted @ 2016-08-22 11:33  西瓜不懂柠檬的酸  Views(1210)  Comments(0Edit  收藏  举报
levels of contents