树形dp——超实用的数据结构

yzc_qiuse

已于 2023-11-11 11:54:37 修改

阅读量82

点赞数 2

分类专栏： c++ 文章标签： c++ 开发语言

于 2023-11-11 11:53:44 首次发布

本文链接：https://blog.csdn.net/yzc_qiuse/article/details/134346436

版权

c++ 专栏收录该内容

18 篇文章 1 订阅

订阅专栏

1. 概念引入

树形 DP，就是以树为模型的动态规划。主要利用树本身带来的子结构（树上的父子关系）来进行状态转移。一般情况下，都是通过子节点的 DP 值来推出父节点的 DP 值。

2. 例题详解

话不多说，我们直接看例题：

2.1 没有上司的舞会

题目链接：没有上司的舞会

2.1.1 题目分析

树形dp的返回值肯定和root相关。
在这里插入图片描述

我们可以根据背包问题的解法来构造树形dp每一维的意义。

dp[root][0]：以root为根节点的子树，当前root节点不来
dp[root][1]：以root为根节点的子树，当前root节点来

状态转移方程：（其中，u是v的子节点）
$d p [u] [0] = Σ ma x (d p [v] [0], d p [v] [1]), v 是 u 的子节点$
$d p [u] [1] = ha pp y [u] + Σ d p [v] [0], v 是 u 的子节点$

状态初值：初值在叶子节点，且不需要手动初始化。

2.1.2 代码

#include<bits/stdc++.h>
using namespace std;
#define endl '\n'
#define int long long
#define For(i, a, b) for(int i = a;i <= b;i++)
typedef pair<int, int> PII;
const int N = 1e4;
vector<int> adj[N]; 
int happy[N], in[N], n, dp[N][2];
void dfs(int u){
    dp[u][1] = happy[u];
    for (int v: adj[u]){
        dfs(v);
        dp[u][1] += dp[v][0];
        dp[u][0] += max(dp[v][0], dp[v][1]);
    }
}
signed main(){
    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    cin >> n;
    For (i, 1, n){
        cin >> happy[i];
    }
    For (i, 1, n - 1){
        int u, v;
        cin >> u >> v;
        adj[v].push_back(u);
        in[u]++;
    }
    int root = 1;
    while (in[root])
        root++;
    dfs(root);
    cout << max(dp[root][0], dp[root][1]) << endl;
    return 0;
}

2.2 选课

2.2.1 题目分析

由于这是一个森林，我们需要构建超级原点。

如果我们单纯从树的角度考虑动态规划，设以 i 为根节点的树选 j 门课程所得到的最大学分为 f（i，j），设虚拟的树根编号为0，学分为0，那么，ans = f (0, n + 1)

如果树根选择1门功课，剩下 j - 1 门功课变成了给他所有儿子如何分配资源的问题，这显然是背包问题。

设前 k 个儿子选修了 x 门课程的最优值为 g(k, x)，则有:
$g(k,x)=\max=\left\{ \begin{aligned} &g(k-1,x) & &第k个儿子不选修 \cr &g(k-1,x-y)+g(son(k),y-1)+a[k] & &第k个儿子选修y门 \cr &这里son(k)指第k个节点的儿子数目 \end{aligned} \right.$

$其中： 0 \leq x \leq j - 1, an s = g (so n (0), n + 1)$

2.2.2 代码

#include<bits/stdc++.h>
using namespace std;
#define endl '\n'
#define int long long
#define For(i, a, b) for(int i = a;i <= b;i++)
typedef pair<int, int> PII;
const int N = 400;
int n, m, w[N], dp[N][N];
vector<int> adj[N];
void dfs(int u){
    for (int v: adj[u]){
        dfs(v);
        for (int tot = m; tot >= 0;tot--){
            For (j, 1, tot){
                dp[u][tot] = max(dp[u][tot], dp[u][tot - j] + dp[v][j]);
            }
        }
    }
    for (int i = m; i > 0;i--){
        dp[u][i] = dp[u][i - 1] + w[u];
    }
}
signed main(){
    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    cin >> n >> m;
    for (int i = 1; i <= n; i++){
        int u, x;
        cin >> u >> x;
        adj[u].push_back(i);
        w[i] = x;
    }
    m += 1;
    dfs(0);
    cout << dp[0][m];
    return 0;
}