2014 Super Training #7 C Diablo III --背包问题(DP)

原题： ZOJ 3769 http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3769

一个带有一些限制的背包问题。

假设在没有限制的情况下，那么定义：dp[i][j]表示在前 i 类物品中，总的Toughness为 j 的时候最大的伤害值。

取到第K类的第x个物品时（属性值为D,T），则有转移方程： dp[K][j+T] = max(dp[K][j+T],dp[K-1][j]+D) .其中j+T超过m时按m算就可以了。

但是有限制如下：

1、对于两个手指的，无论是只装备一根手指，还是装备了两只，都用手指这一类来表示，那么，所有手指装备本身当作只装备一根手指，装备两只的两两枚举一下

2、对于Weapon和Shield两种道具以及Two-Handed，我们还是把它们当成一种来处理，首先各自肯定当成一件物品，然后就是枚举Weapon和Shield的搭配了当成一种，将这些归为一类

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <algorithm>
#include <string>
#include <vector>
using namespace std;
#define N 50007

string god[] = {"Head", "Shoulder", "Neck", "Torso", "Hand", "Wrist", "Waist", "Legs", "Feet", "Finger", "Shield", "Weapon", "Two-Handed"};

struct Good
{
    int damag,tough;
    Good(int _damg,int _togh)
    {
        damag = _damg;
        tough = _togh;
    }
    Good(){}
};

vector<Good> G[];

int getnum(string ka)
{
    for(int i=;i<;i++)
    {
        if(god[i] == ka)
            return i;
    }
}

int dp[][N];

int main()
{
    int i,j,k;
    string ss;
    int t,n,m;
    int damag,tough;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        for(i=;i<=;i++)
            G[i].clear();
        for(i=;i<n;i++)
        {
            cin>>ss;
            scanf("%d%d",&damag,&tough);
            k = getnum(ss);
            G[k].push_back(Good(damag,tough));
            if(k ==  || k == )  //weapon or sheild,一并算在Two_handed里面
                G[].push_back(Good(damag,tough));
        }
        //枚举weapon and sheild 's combination
        for(i=;i<G[].size();i++)
            for(j=;j<G[].size();j++)
                G[].push_back(Good(G[][i].damag+G[][j].damag,G[][i].tough+G[][j].tough));
        G[].clear();
        G[].clear();
        //G[10] 存放finger所有的情况（单独和组合）
        for(i=;i<G[].size();i++)
        {
            G[].push_back(G[][i]);
            for(j=i+;j<G[].size();j++)
                G[].push_back(Good(G[][i].damag+G[][j].damag,G[][i].tough+G[][j].tough));
        }
        G[].clear();  //注意清空，情况都加到G[10]里面去了
        memset(dp,-,sizeof(dp));
        dp[][] = ;
        int T,D;
        for(i=;i<G[].size();i++)
        {
            Good g = G[][i];
            T = min(g.tough,m);
            dp[][T] = max(dp[][T],g.damag);
        }
        for(k=;k>=;k--)
        {
            for(i=;i<=m;i++)
            {
                dp[k][i] = max(dp[k][i],dp[k+][i]);
                if(dp[k+][i] == -)
                    continue;
                for(j=;j<G[k].size();j++)
                {
                    Good g = G[k][j];
                    T = min(g.tough+i,m);
                    D = g.damag+dp[k+][i];
                    dp[k][T] = max(dp[k][T],D);
                }
            }
        }
        printf("%d\n",dp[][m]);
    }
    return ;
}