HDU 3535

  这是一道综合性的背包问题。题目给出了多组工作,每组工作的选择规则不同,有些组至少要选一项,有些组最多选一项,有些组任意选。

  这道题花了很长时间,需要深入理解状态转移才能够明白。数组dp[i][j],表示第i组,时间剩余为j时的快乐值。每得到一组工作就进行一次DP,所以dp[i]为第i组的结果。下面对三种情况进行讨论。

  第一类,至少选一项,即必须要选,那么在开始时,对于这一组的dp的初值,应该全部赋为负无穷,这样才能保证不会出现都不选的情况。状态转移方程为dp[i][k]=max{ dp[i][k],dp[i-1][k-cost[j]]+val[k],dp[i][k-cost[j]]+val[j] }。dp[i][k]是不选择当前工作;dp[i-1][k-cost[j]]+val[k]是选择当前工作,但是是第一次在本组中选,由于开始将该组dp赋为了负无穷,所以第一次取时,必须由上一组的结果推知,这样才能保证得到全局最优解;dp[i][k-cost[j]]+val[j]表示选择当前工作,并且不是第一次取。

  第二类,最多选一项,即要么不选,一旦选,只能是第一次选。所以状态转移方程为dp[i][k]=max{ dp[i][k],dp[i-1][k-cost[j]]+val[k]}。由于要保证得到全局最优解,所以在该组DP开始以前,应该将上一组的DP结果先复制到这一组的dp[i]数组里,因为这一组的数据是在上一组数据的基础上进行更新的。

  第三类,任意选,即不论选不选,选几次都可以,显然状态转移方程为dp[i][k]=max{ dp[i][k],dp[i-1][k-cost[j]]+val[k],dp[i][k-cost[j]]+val[j] }。同样要保证为得到全局最优解,先复制上一组解。

  下面是代码

#include<stdio.h>
#include<string.h>
#define INF 1000000
#define MAX_LIMT 110
int get_max(int,int);
int main()
{
    int n,t;
    while(scanf("%d%d",&n,&t)!=EOF)
    {
    int i,j,k,dp[MAX_LIMT][MAX_LIMT];
    memset(dp,0,sizeof(dp));
    for(i=1;i<=n;i++)
    {
        int m,s,cost[MAX_LIMT],val[MAX_LIMT];
        scanf("%d%d",&m,&s);
        for(j=1;j<=m;j++)
        {
            scanf("%d%d",&cost[j],&val[j]);
        }
        if(s==0)
        {
            for(j=0;j<=t;j++)
            {
                dp[i][j]=-INF;
            }
            for(j=1;j<=m;j++)
            {
                for(k=t;k>=cost[j];k--)
                {
                    dp[i][k]=get_max( dp[i][k],dp[i][k-cost[j]]+val[j]);
                    dp[i][k]=get_max( dp[i][k],dp[i-1][k-cost[j]]+val[j]);
                }
            }
        }
        else if(s==1)
        {
            for(j=0;j<=t;j++)
            {
                dp[i][j]=dp[i-1][j];
            }
            for(j=1;j<=m;j++)
            {
                for(k=t;k>=cost[j];k--)
                {
                    dp[i][k]=get_max(dp[i][k],dp[i-1][k-cost[j]]+val[j]);
                }
            }
        }
        else
        {
            for(j=0;j<=t;j++)
            {
                dp[i][j]=dp[i-1][j];
            }
            for(j=1;j<=m;j++)
            {
                for(k=t;k>=cost[j];k--)
                {
                    dp[i][k]=get_max( dp[i][k],dp[i][k-cost[j]]+val[j]);
                    dp[i][k]=get_max( dp[i][k],dp[i-1][k-cost[j]]+val[j]);
                }
            }
        }
    }
    dp[n][t]=get_max(dp[n][t],-1);
    printf("%d\n",dp[n][t]);
    }
    return 0;
}
int get_max(int x,int y)
{
    return x>y?x:y;
}
posted @ 2012-07-26 23:25  等待电子的砹  阅读(1623)  评论(0编辑  收藏  举报