POJ1014 多重背包(二进制拆分)

题目链接
题目大意:给出6个数,分别代表价值为1~6的6种物品的数量,求能否在价值相同的情况下平分这些物品。
思路:多重背包。一开始拆成01背包来做,写了三层循环,很显然结果TLE了。后来上网看了一下题解,大致意思是把每一种物品的数量拆成1,2,4,8,……这样的2的n次方数,比如将12拆成1,2,4,5四个数,这样可以用这些数字来拼凑成任意一个所需要背包的数。

#include"cstdio"
#include"iostream"
#include"cstring"
#include"algorithm"
#include"cmath"
using namespace std;
int x[10];//初始输入
int dp[60011];
int y[20005];//拆分后的物品价值
int max(int a,int b)
{
    return a>b?a:b;
}
int main()
{
    int cas=1;
    //暴力输入
    while(~scanf("%d%d%d%d%d%d",&x[1],&x[2],&x[3],&x[4],&x[5],&x[6]),x[1]||x[2]||x[3]||x[4]||x[5]||x[6])
    {
        int sum=x[1]+x[2]*2+x[3]*3+x[4]*4+x[5]*5+x[6]*6; 
        if(sum&1)//奇数价值不可能平分
        {
            printf("Collection #%d:\nCan't be divided.\n\n",cas++);
            continue;
        }
        int cnt=0;
        sum=sum/2;
        memset(dp,0,sizeof dp);
        //拆分写搓WA了很多遍
        for(int i=1;i<=6;i++)
        {
            int k=1;
            while(k<x[i])
            {
                y[cnt++]=k*i;//拆分后当作新的物品,有自己的价值
                x[i]-=k;
                k*=2;
            }
            y[cnt++]=x[i]*i;
        }
        for(int i=0;i<cnt;i++)//背包
        {
            for(int k=sum;k>=y[i];k--)
                dp[k]=max(dp[k],dp[k-y[i]]+y[i]);//状态转移方程
        }
        if(dp[sum]==sum)
            printf("Collection #%d:\nCan be divided.\n\n",cas++);
        else
            printf("Collection #%d:\nCan't be divided.\n\n",cas++);
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值