POJ1014 多重背包

今天开始DP了,第一道多重背包,模板题,还是想了很久,太弱了……..
题意:一行给出6个数,表示从1到6,6个不同权值的个数.求能否将所有权值不分割的分成权值相等的两份.
首先总权值为奇数的肯定不符合要求,接下来就是一个多重背包了.
多重背包的二进制拆分我的理解大概是;
首先将一个数按二进制拆分之后可以用拆分后的数相加得到小于或等于这个数的任意一个数.拆分之后得到许多小堆,接下来就只需要以堆为单位做一次0-1背包就可以了.

#include <iostream>
#include <cstring>
#include <math.h>
using namespace std;

struct node
{
    int num;
    int value;
}div[20000];

int n[7];
int dp[60005];

int main()
{
    int cnt=1;
    while(true)
    {
        int i,j;
        int sum=0;
        memset(dp,0,sizeof(dp));
        for(i=1;i<=6;i++)
        {
            cin>>n[i];
            sum+=n[i]*i;
        }
        if(sum==0) break;
        if(sum%2!=0)
        {
            cout<<"Collection #"<<cnt++<<":"<<endl;
            cout<<"Can't be divided."<<endl<<endl;
            continue;
        }
        sum=sum/2;
        int kcnt=0;
        for(i=1;i<=6;i++)
        {
            int m=n[i];
            int k=1;
            while(k<=m)
            {
                div[kcnt].num=k;
                div[kcnt++].value=i;
                m-=k;
                k*=2;
            }
            div[kcnt].num=m;
            div[kcnt++].value=i;
        }
        for(i=0;i<kcnt;i++)
        {
            int tem=div[i].num*div[i].value;
            for(j=sum;j>=tem;j--)
            {
                dp[j]=max(dp[j],dp[j-tem]+tem);
            }
        }
        if(dp[sum]==sum)
        {
            cout<<"Collection #"<<cnt++<<":"<<endl;
            cout<<"Can be divided."<<endl<<endl;
        }
        else
        {
            cout<<"Collection #"<<cnt++<<":"<<endl;
            cout<<"Can't be divided."<<endl<<endl;
        }
    }
    return 0;
}