LightOJ 1147 Tug of War(状态压缩)

题目链接:http://www.lightoj.com/volume_showproblem.php?problem=1147

题意:将n个数分成两个集合,两个集合的大小相差最多为1的前提下使得两个集合的数字之和的差尽量小?输出两个集合的数字之和。

思路:f[j]表示得到j可由几个数字得到,比如f[j]=1110(二进制)表示j可由1个或2个或3个数字得到。。。由于每个集合最多50个数,long long正好够了。。好巧妙的思路啊。。。。

#include <iostream>

#include <cstdio>

#include <cstring>

#define i64 long long

using namespace std;





int C,num=0;

int n,a[105];

i64 f[50005];







int main()

{

    for(scanf("%d",&C);C--;)

    {

        scanf("%d",&n);

        int i,j,sum=0,t;

        for(i=1;i<=n;i++) scanf("%d",&a[i]),sum+=a[i];

        t=sum>>1;

        memset(f,0,sizeof(f));

        f[0]=1;

        for(i=1;i<=n;i++) for(j=t;j>=a[i];j--) f[j]|=f[j-a[i]]<<1;

        int x=(n+1)>>1,y=n>>1;

        i64 p=1;

        for(j=t;j>0;j--)

        {

            if((f[j]&(p<<x))||(f[j]&(p<<y))) break;

        }

        printf("Case %d: %d %d\n",++num,j,sum-j);

    }

    return 0;

}

  

你可能感兴趣的:(war)