HDU-4044-treeDP+分组背包

我觉得这题挺难想的。反正我是想了好久。

题意：一棵树（编号1-n），1是敌人出口（只有一个敌人）。叶子节点是我军。在节点处装大炮。每个节点有 k 种大炮选择，可是每个节点最多装一个大炮。每个大炮有一个花费和一个威力值。你一共有m钱，怎么能使威力值最大。敌人去攻击哪个我军是任意的，所以最大威力是每条路值和的最小值。大炮放在叶子节点也是管用的。

解题思路：每个节点最多装一个大炮，所以是分组背包 。 树与子树之间的分组背包，树与树根节点之间的分组背包。看代码吧。

const int inf = 1<<30;
const int N = 1009;
const int C = 209;

int dp[N][C]; //dp[i][j] i为根的子树 花费j最大值
int hp[N][C]; //hp[i][j] 第i节点 花费j最大值
int n, m;
vector<int>v[N];

void dfs(int s, int pre)
{
    int len = v[s].size();
    int i, j, k, ss;
    if(len==1 && s!=1) //叶子节点
    {
        for(i=0; i<=m; i++) dp[s][i] = hp[s][i];
        return;
    }
    for(i=0; i<=m; i++) dp[s][i] = inf;
    for(i=0; i<len; i++)
    {
        ss = v[s][i];
        if(ss==pre) continue;
        dfs(ss, s);
        for(j=m; j>=0; j--) //树与子树之间的分组背包，子树花费和是j
        {
            int t = 0;
            for(k=0; k<=j; k++) //ss子树花费是k
                t = max(t, min(dp[s][j-k], dp[ss][k])); //各个子树威力值最小的
            dp[s][j] = t;
        }
    }

    for(j=m; j>=0; j--) //树与该树根节点之间的分组背包
        for(k=0; k<=j; k++)
            dp[s][j] = max(dp[s][j], dp[s][j-k]+hp[s][k]);
}
int main()
{
    int t;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d", &n);
        for(int i=1; i<=n; i++) v[i].clear();
        memset(hp, 0, sizeof(hp));
        for(int i=1; i<n; i++)
        {
            int x, y;
            scanf("%d%d", &x, &y);
            v[x].push_back(y);
            v[y].push_back(x);
        }
        scanf("%d", &m);
        for(int i=1; i<=n; i++)
        {
            int k, x, y;
            scanf("%d", &k);
            for(int j=0; j<k; j++)
            {
                scanf("%d%d", &x, &y);
                hp[i][x] = max( hp[i][x] ,y );
            }
        }
        for(int i=1; i<=n; i++)
            for(int j=1; j<=m; j++)
                hp[i][j] = max(hp[i][j], hp[i][j-1]);
        dfs(1, -1);

        printf("%d\n", dp[1][m]);
    }
	return 0;
}