hdu 4044 GeoDefense 树形DP+分组背包

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4044

题意：一棵树，n个节点，叶子节点为基地。在节点处可以放大炮，每个节点都有ki种大炮，每种大炮有一个花费和一个攻击力。每个节点最多只能放一个大炮。一个怪物从节点1出发，可能走任意路径，需要在其到达基地前消灭。求金额m所能消灭怪物的最大HP值。

根据题意，我们要使最薄弱的基地的攻击力最大化。由于每个节点最多只能放一个大炮，所以我们先求出节点i给定花费j的攻击力最大值，用p[i][j]来表示。

我们设以节点t为根节点花费j的最大攻击力d[t][j]。现在分析节点r：

1.t是叶子节点，只需考虑自身防御情况，所以d[t][j]=p[t][j]。

2.t不是叶子节点。先考虑不在节点t上放大炮，遍历子树，求出花费j子树最小值的最大组合。然后考虑分配k花费给节点t，j-k花费给子树的最大组合。

#include <iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<vector>
#define N 1100
#define M 220
#define INF 0x7ffffff
using namespace std;

vector<int> r[N];
int n,m,d[N][M],v[N],p[N][M];

void dfs(int t)
{
    v[t]=1;
    if(t!=1&&r[t].size()==1)//t为树叶
    {
        for(int j=0;j<=m;j++)   d[t][j]=p[t][j];
        return ;
    }
    for(int j=0;j<=m;j++)   d[t][j]=INF;//初始化
    for(int i=0;i<r[t].size();i++)//求只分配给子树的情况
    {
        int u=r[t][i];
        if(v[u])    continue;
        dfs(u);
        for(int j=m;j>=0;j--)
        {
            int maxc=0;
            for(int k=0;k<=j;k++)
                maxc=max(maxc,min(d[t][j-k],d[u][k]));//子树中的最小值的最大组合
            d[t][j]=maxc;
        }
    }
    for(int j=m;j>=0;j--)
        for(int k=0;k<=j;k++)
            d[t][j]=max(d[t][j],d[t][j-k]+p[t][k]);//更新，子树分配j-k花费，根节点分配k花费
}

int main()
{
    int T;
    cin>>T;
    while(T--)
    {
        cin>>n;
        for(int i=0;i<=n;i++)   r[i].clear();
        for(int i=1;i<n;i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            r[x].push_back(y);
            r[y].push_back(x);
        }
        cin>>m;
        memset(p,0,sizeof(p));
        for(int i=1;i<=n;i++)
        {
            int k,c,w;
            scanf("%d",&k);
            for(int j=0;j<k;j++)
            {
                scanf("%d%d",&c,&w);
                p[i][c]=max(p[i][c],w);
            }
            for(int j=1;j<=m;j++)
                p[i][j]=max(p[i][j],p[i][j-1]);//求出节点i处j花费的最大防御
        }
        memset(d,0,sizeof(d));
        memset(v,0,sizeof(v));
        dfs(1);
        cout<<d[1][m]<<endl;
    }
}