hdu3033 分组背包变形

来源:互联网 发布:地信算法及代码 编辑:程序博客网 时间:2024/05/19 17:25

http://acm.hdu.edu.cn/showproblem.php?pid=3033

借鉴了别人的思路:

设计状态dp[i][j]代表前i组容量为j的最大价值。由于一组里面有多个物品,所以状态转移可以是前一组少取一个,即dp[i-1][p-g[i][j].v]+g[i][j].w,也可以是当前组之前去过的少取一种,即dp[i][p-g[i][j].v]+g[i][j].w。

    网上有些解题报告是错误的解法,这题dp初始化的时候要初始化为负无穷,因为这题要求的是上一组恰好达到的状态才能转移到这一组来,因为每一组至少得去一个。当然可以初始化为-1,每次判断一下是不是-1就行。

#include <iostream>#include<cstdio>#include<cstring>using namespace std;struct node{    int v,w;};node g[15][105];int num[15];int dp[15][10005];int main(){    int m,n,k;    while(scanf("%d%d%d",&n,&m,&k)!=EOF)    {        memset(num,0,sizeof(num));        int i;        for(i=1;i<=n;i++)        {            int j,a,b;            scanf("%d%d%d",&j,&a,&b);            g[j][num[j]].w=a;            g[j][num[j]].v=b;            num[j]++;        }        int j;        memset(dp,-1,sizeof(dp));        memset(dp[0],0,sizeof(dp[0]));        for(i=1;i<=k;i++)        {            for(j=0;j<num[i];j++)            {                int p;                for(p=m;p>=g[i][j].w;p--)                {                    if(dp[i][p-g[i][j].w]!=-1)                        dp[i][p]=max(dp[i][p-g[i][j].w]+g[i][j].v,dp[i][p]);                    if(dp[i-1][p-g[i][j].w]!=-1)                        dp[i][p]=max(dp[i][p],dp[i-1][p-g[i][j].w]+g[i][j].v);                }            }        }        if(dp[k][m]==-1)            printf("Impossible\n");        else            printf("%d\n",dp[k][m]);    }    return 0;}


原创粉丝点击