AcWing
  • 首页
  • 课程
  • 题库
  • 更多
    • 竞赛
    • 题解
    • 分享
    • 问答
    • 应用
    • 校园
  • 关闭
    历史记录
    清除记录
    猜你想搜
    AcWing热点
  • App
  • 登录/注册

经典的树形DP题型

作者: 作者的头像   acwing_98698 ,  2025-05-10 18:07:33 · 广东 ,  所有人可见 ,  阅读 7


0


最小支配集问题

题意

给定一棵树,选择一个点后可以覆盖该点以及所有与之相邻的点,求选择最少的点使整棵树被覆盖。

思路

定义三种状态:

  • dp[u][0]:选择u节点后,使整棵子树被覆盖的最少点集
  • dp[u][1]:不选u节点但被子节点覆盖,使整棵子树被覆盖的最少点集
  • dp[u][2]:不选u节点但被父节点覆盖,使整棵子树被覆盖的最少点集

状态转移:

  • 选择u节点,则其子节点可选可不选,取最少点集方案求和。

$$ dp[u][0] = \sum_{v \in son[u]} min(dp[v][0], dp[v][1],dp[v][2]); $$

  • 不选u节点但被子节点覆盖,意味着有一个子节点一定被选择,其他的子节点可选可不选取最少点集方案求和。

$$ dp[u][1] = min(dp[u][1], dp[x][0] + \sum_{v \in son[u] \ \&\& \ v \neq x} min(dp[v][0], dp[v][1]) ); $$

变换公式:

$$ dp[u][1] = min(dp[u][1], dp[x][0] + \sum_{v \in son[u]} min(dp[v][0], dp[v][1]) - min(dp[x][0], dp[x][1])); $$

其中 $\sum_{v \in son[u]} min(dp[v][0], dp[v][1])$ 是定值,决定大小的是 $dp[x][0] - min(dp[x][0], dp[x][1])$。

  • 不选u节点但被父节点覆盖,则其子节点可选可不选,但不选时无法被u节点覆盖,取最少的点集方案求和。

$$ dp[u][2] = \sum_{v \in son[u]} min(dp[v][0], dp[v][1]); $$

这里可以发现 dp[u][2] 的值就是上面说的定值,所以 dp[u][1] 在计算过程中只需计算 $dp[x][0] - min(dp[x][0], dp[x][1])$ 的最小值。

初始化DP数组

当u节点被选择时,点集数加一,同时 dp[u][1] 要计算 $dp[x][0] - min(dp[x][0], dp[x][1])$ 的最小值,需要初始化为最大值。

最后,因为根节点是不可能被父节点覆盖的,所以答案取 $min(dp[1][1], dp[1][0])$。

参考代码

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

typedef long long ll;

const int INF = 1e9;
const int N = 1e5 + 10;

int n;
vector<int> g[N];
ll dp[N][3];

void dfs(int u, int fa)
{
    dp[u][0] = 1;
    dp[u][1] = INF;
    for (auto v : g[u]) {
        if (v == fa) continue;
        dfs(v, u);

        dp[u][0] += min({dp[v][0], dp[v][1], dp[v][2]});
        dp[u][1] = min(dp[u][1], dp[v][0] - min(dp[v][0], dp[v][1]));
        dp[u][2] += min(dp[v][0], dp[v][1]);
    }
    dp[u][1] = dp[u][2];
}

int main()
{
    cin >> n;
    for (int i = 1; i < n; i ++) {
        int u, v;
        cin >> u >> v;
        g[u].push_back(v);
        g[v].push_back(u);
    }

    dfs(1, -1);

    cout << min(dp[1][0], dp[1][1]);

    return 0;
}

推荐例题: P2899 [USACO08JAN] Cell Phone Network G

K度最小支配集问题

题意

给定一棵树,选择一个点后可以覆盖所有与之最短距离不超过 k 的点,求选择最少的点覆盖整棵树。

思路

定义状态

  • dp[u][0]:记录以节点u为根的子树中,未被覆盖的最远距离(即需要被覆盖的需求)
  • dp[u][1]:记录以节点u为根的子树中,最近的覆盖点到u的距离(即提供的覆盖能力)

状态转移

如果最近的覆盖点到u的距离超过k,则可能需要将其未被覆盖距离设为0。
如果未被覆盖的最远距离加上最近的覆盖距离不超过k,说明该节点已被覆盖,重置状态。
若未被覆盖的最远距离等于k,则必须在此节点放置覆盖点,并更新状态。

最后,若根节点的未被覆盖距离非负,说明需要额外覆盖点。

参考代码

#include <iostream>
#include <vector>

using namespace std;

const int INF = 1e9;
const int N = 1e5 + 10;

int n, k, t, ans;
vector<int> g[N];
int dp[N][2];

void dfs(int u, int fa)
{
    dp[u][0] = -INF;
    dp[u][1] = INF;
    for (auto v : g[u]) {
        if (v == fa) continue;
        dfs(v, u);

        dp[u][0] = max(dp[u][0], dp[v][0] + 1);
        dp[u][1] = min(dp[u][1], dp[v][1] + 1);
    }
    if (dp[u][1] > k) dp[u][0] = max(0, dp[u][0]);
    if (dp[u][0] + dp[u][1] <= k) dp[u][0] = -INF;
    if (dp[u][0] == k) dp[u][0] = -INF, dp[u][1] = 0, ans ++;
}

int main()
{
    cin >> n >> k;
    for (int i = 1; i < n; i ++) {
        int u, v;
        cin >> u >> v;
        g[u].push_back(v);
        g[v].push_back(u);
    }

    dfs(1, -1);

    if (dp[1][0] >= 0) ans ++;
    cout << ans;

    return 0;
}

推荐例题: P3942 将军令
P2279 [HNOI2003] 消防局的设立

0 评论

App 内打开
你确定删除吗?
1024
x

© 2018-2025 AcWing 版权所有  |  京ICP备2021015969号-2
用户协议  |  隐私政策  |  常见问题  |  联系我们
AcWing
请输入登录信息
更多登录方式: 微信图标 qq图标 qq图标
请输入绑定的邮箱地址
请输入注册信息