Shopping Offers POJ - 1170

http://poj.org/problem?id=1170

将每一种交易当作一件物品 每件物品有无数件 问题就转换为完全背包问题

(000)(000)(000)(000)(000) 最多5种商品 每种商品用一个八进制数表示其数量 总的状态不超过2^15 以此作为背包容量 而每次交易即每种物品的重量也是对每种商品分开算

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=10;
const int maxq=2e2+10;
const int maxm=1e3+10;
const int maxs=1e5+10;

struct node
{
    int p;
    int k[maxn];
};

node pre[maxq];
int dp[maxs];
int c[maxn],k[maxn],p[maxn],book[maxm];
int tot,n,q;

int main()
{
    int i,j,kk,sta,tc,tk;
    scanf("%d",&n);
    for(i=0;i<n;i++){
        scanf("%d%d%d",&c[i],&k[i],&p[i]);
        book[c[i]]=i;
    }
    for(i=0;i<n;i++){
        for(j=1;j<=k[i];j++){
            tot++;
            pre[tot].k[i]=j;
            pre[tot].p=j*p[i];
        }
    }
    scanf("%d",&q);
    for(i=tot+1;i<=tot+q;i++){
        scanf("%d",&sta);
        while(sta--){
            scanf("%d%d",&tc,&tk);
            tc=book[tc];
            pre[i].k[tc]=tk;
        }
        scanf("%d",&pre[i].p);
    }
    q=tot+q;
    for(i=0;i<n;i++){
        for(j=1;j<=k[i];j++){
            q++;
            pre[q].k[i]=j;
            pre[q].p=j*p[i];
        }
    }
    /*
    for(i=1;i<=q;i++){
        printf("***%d %d***\n",i,pre[i].p);
        for(j=0;j<n;j++){
            printf("%d ",pre[i].k[j]);
        }
        printf("\n");
    }
    */
    memset(dp,0x3f,sizeof(dp));
    dp[0]=0;
    tot=1<<(3*n);
    //for(j=0;j<tot;j++) printf("%d\n",dp[j]);
    for(i=1;i<=q;i++){
        //printf("***%d %d***\n",i,pre[i].p);
        for(j=0;j<tot;j++){
            sta=j;
            for(kk=0;kk<n;kk++){
                if(sta%8>=pre[i].k[kk]){
                    sta/=8;
                }
                else break;
            }
            if(kk==n){
                //printf("*%d*\n",j);
                sta=j;
                for(kk=0;kk<n;kk++){
                    sta-=pre[i].k[kk]*(1<<(3*kk));
                }
                dp[j]=min(dp[j],dp[sta]+pre[i].p);
            }
        }
    }
    sta=0;
    for(i=0;i<n;i++){
        sta+=k[i]*(1<<(3*i));
    }
    printf("%d\n",dp[sta]);
    return 0;
}