2020 ICPC南京 M.Monster Hunter(树形背包)

题目大意

  给你一棵树,你可以去掉i个点(i∈[0,n]i\in[0,n]i[0,n]),然后计算剩余结点的贡献,每个结点的贡献为它本身的价值加上它所有儿子的价值。当然你需要合理安排删去的i个结点使贡献最小化。输出每个i对应的贡献值。

解体思路

  树形dp,由于结点的贡献与周边的结点有关,所以还需要再开一维状态
状态设计
dp[0][i][j]dp[0][i][j]dp[0][i][j] 表示删掉结点i,并且子树中留下j个结点的价值。
dp[1][i][j]dp[1][i][j]dp[1][i][j] 表示留下结点i,并且子树中留下j-1个结点的价值。
状态转移
dp[0][i][j+k]=min(f[0][i][j+k],f[0][i][j]+min(f[0][son][k],f[1][son][k]))dp[0][i][j+k] = min(f[0][i][j+k],f[0][i][j]+min(f[0][son][k],f[1][son][k]))dp[0][i][j+k]=min(f[0][i][j+k],f[0][i][j]+min(f[0][son][k],f[1][son][k]))
dp[1][i][j+k]=min(f[1][i][j+k],f[1][i][j]+min(f[0][son][k],f[1][son][k]+a[son]))dp[1][i][j+k] = min(f[1][i][j+k],f[1][i][j]+min(f[0][son][k],f[1][son][k] + a[son]))dp[1][i][j+k]=min(f[1][i][j+k],f[1][i][j]+min(f[0][son][k],f[1][son][k]+a[son]))
其中a数组存储的是结点的价值
边界条件
f[0][i][0]=0f[0][i][0] = 0f[0][i][0]=0
f[1][i][1]=a[i]f[1][i][1]=a[i]f[1][i][1]=a[i]

代码实现

注意dp时边界的确定,否则复杂度不是O(n)的
循环中第一维倒序枚举的原因是运用了滚动数组

#include <bits/stdc++.h>
using namespace std;
 
const int MAXN = 2e3 +5;
const int MAXM = 4e3 +5;
#define ll long long
 
const ll inf = 1e18;
 
int head[MAXN], nxt[MAXM], to[MAXM], sze[MAXN], a[MAXN], n, tot;
inline void AddEdge(int u, int v) {
     nxt[++tot] = head[u], to[head[u] = tot] = v;
}

ll f[2][MAXN][MAXN];
void dfs(int u) {
    f[0][u][0] = 0;
    f[1][u][1] = a[u];

    for(int e = head[u]; e; e=nxt[e])
        dfs(to[e]);

    sze[u] = 1;
    for (int e = head[u]; e; e =  nxt[e]) {
        for (int j = sze[u]; j >= 0; j--) {
            for (int k = sze[to[e]]; k >= 0; k--) {
                f[0][u][j + k] = min(f[0][u][j + k], f[0][u][j] + min(f[0][to[e]][k], f[1][to[e]][k]));
                f[1][u][j + k] = min(f[1][u][j + k], f[1][u][j] + min(f[0][to[e]][k], f[1][to[e]][k] + a[to[e]]));
            }
        }
        sze[u] += sze[to[e]];
    }
}
 
int main() {
    int T; scanf("%d", &T);
    while (T--) {
        scanf("%d", &n);
        for (int i = 1; i <= n; i++) head[i] = 0;
        tot = 0;
        for (int i = 2; i <= n; i++) {
            int u;
            scanf("%d", &u);
            AddEdge(u, i);
        }
        for (int i = 1; i <= n; i++)
            scanf("%d", a + i);
 
        for (int i = 0; i <= n; i++)
            for (int j = 0; j <= n; j++)
                f[0][i][j] = f[1][i][j] = inf;
                
        dfs(1);
        for (int i = 0; i <= n; i++)
            printf("%lld%c", min(f[0][1][n - i], f[1][1][n - i]), " \n"[i == n]);
    }
 	return 0;
}
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值