DSU on Tree（树上启发式合并）学习笔记

已于 2024-01-26 22:56:30 修改
阅读量931 收藏 22
点赞数 27
分类专栏： DP 文章标签： 学习 笔记 算法 c++
于 2024-01-26 22:51:07 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/jinniuzhiguan/article/details/135875692
版权

引入

有这样一个问题：给定一棵结点数为 n n n 的树，求其长度刚好为 K K K 的简单路径数量， n n n 和 K K K 都是 1 0 5 10^5 105 级别。

此题有多种解法，如点分治，长链剖分，最暴力的一种就是直接 dp，设 f u , i f_{u,i} fu,i​ 表示 u u u 的子树中到 u u u 长度为 i i i 的路径数量，然后暴力合并并计算答案。时间复杂度 O ( n 2 ) O(n^2) O(n2)。

这里介绍一种 O ( n log ⁡ n ) O(n\log n) O(nlogn) 的做法。

DSU on Tree

首先想想我们需要维护什么，如果对于当前维护的 u u u 为根的联通块，知道 f u , i f_{u,i} fu,i​，那么就可以将 u u u 的一个不在联通块内的子树内所有点，和联通块内的点构成点对并计算答案，然后合并到联通块。

什么意思呢，首先是计算答案，若子树内一个点到 u u u 的距离为 d d d，那联通块内对应的点到 u u u 的距离就是 K − d K-d K−d，如果我们知道联通块内到 u u u 距离为 K − d K-d K−d 的点的数量的话，就能将这个数量加到答案里。计算完后，把这个子树合并到联通块中，同时维护 f u f_{u} fu​。

不过到一个点的距离不好维护，可以改成维护深度，记当前联通块深度为 i i i 的点的数量为 c n t i cnt_i cnti​，这样一减就是两个点的距离。

但这样计算答案并合并，就好把子树遍历一遍，是 O ( n 2 ) O(n^2) O(n2) 的，这时就需要用 dsu on tree。

我们可以先进行重链剖分，在递归计算答案时，对于轻儿子维护出来的 c n t cnt cnt，直接丢掉，对于重儿子维护出来的 c n t cnt cnt，把它留着。先将重儿子的子树作为初始联通块，然后把所有轻儿子的子树依次对联通块计算答案，接着合并。没错，遍历一遍轻儿子的子树，暴力合并。

众所周知，重链剖分后每个点到根节点的轻边数量是 O ( log ⁡ n ) O(\log n) O(logn) 的，每次合并是 O ( n ) O(n) O(n)，所以总复杂度是 O ( n log ⁡ n ) O(n\log n) O(nlogn) 的！

看起来有点抽象？这里有道例题。

例题 1：[CF161D] Distance in Tree

这题数据范围并不强，但也可以用来练手，题目就是上面所说的。代码如下：

/*
    Program: CF161D.cpp
    Author: 1l6suj7
    DateTime: 2024-01-26 18:04:43
    Description:
*/

#include <bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define lp(i, j, n) for(int i = j; i <= n; ++i)
#define dlp(i, n, j) for(int i = n; i >= j; --i)
#define mst(n, v) memset(n, v, sizeof(n))
#define mcy(n, v) memcpy(n, v, sizeof(v))
#define INF 1e18
#define MAX4 0x3f3f3f3f
#define MAX8 0x3f3f3f3f3f3f3f3f
#define pii pair<int, int>
#define pll pair<ll, ll>
#define co(x) cerr << (x) << ' '
#define cod(x) cerr << (x) << endl
#define fi first
#define se second
#define eps 1e-8
#define lc(x) ((x) << 1)
#define rc(x) ((x) << 1 ^ 1)
#define pb(x) emplace_back(x)

using namespace std;

const int N = 50010, M = 510;

struct edge { int v, nxt; } E[N << 1];
int en, hd[N];

void add(int u, int v) { E[++en] = { v, hd[u] }, hd[u] = en; }

int n, K;
ll ans;

int sz[N], son[N], dep[N], L[N], R[N], nd[N], dfc, cnt[N];
/*
sz：子树大小
son：重儿子
dep：到根节点距离
L：子树内最小 dfn
R: 子树内最大 dfn
nd：dfn 对应的结点
dfc：记录当前 dfn
cnt[i]：当前联通块深度为 i 的结点数量
*/
void dfs1(int u, int fa) {	// 预处理
    sz[u] = 1, L[u] = R[u] = ++dfc, nd[dfc] = u;
    for(int i = hd[u]; i; i = E[i].nxt) {
        int v = E[i].v;
        if(v == fa) continue;
        dep[v] = dep[u] + 1, dfs1(v, u), sz[u] += sz[v], R[u] = max(R[u], R[v]);
        if(sz[v] > sz[son[u]]) son[u] = v;
    }
}

void calc(int grd, int u) {		// 计算子树内的一个点与联通块内点构成路径的答案
    int t = dep[grd] + K - dep[u] + dep[grd];	// 联通块内点的深度
    if(t >= 0 && t <= n) ans += cnt[t];
}

void Add(int u) { ++cnt[dep[u]]; }	// 向联通块内添加点

void Del(int u) { --cnt[dep[u]]; }	// 删除点

void dfs2(int u, int fa, bool kp) {		// kp 表示最后是否丢掉维护的东西
    for(int i = hd[u]; i; i = E[i].nxt) {	// 先计算轻儿子
        int v = E[i].v;
        if(v == fa || v == son[u]) continue;
        dfs2(v, u, 0);
    }
    if(son[u]) dfs2(son[u], u, 1);		// 计算重儿子
    for(int i = hd[u]; i; i = E[i].nxt) {
        int v = E[i].v;
        if(v == fa || v == son[u]) continue;
        lp(j, L[v], R[v]) calc(u, nd[j]);	// 先记录答案
        lp(j, L[v], R[v]) Add(nd[j]);		// 暴力合并
    } calc(u, u), Add(u);
    if(kp) return;
    lp(i, L[u], R[u]) Del(nd[i]);	// 丢掉维护的东西
}

signed main() {
    // freopen(".in", "r", stdin);
    // freopen(".out", "w", stdout);
#ifndef READ
    ios::sync_with_stdio(false);
    cin.tie(0);
#endif
    cin >> n >> K;
    int u, v;
    lp(i, 1, n - 1) cin >> u >> v, add(u, v), add(v, u);
    dfs1(1, 0), dfs2(1, 0, 0);
    cout << ans << endl;
    return 0;
}

例题 2：P3806 【模板】点分治 1

要求树上距离为 k k k 的点对是否存在，只不过有边权，深度会很大，所以用 map 来维护（实际上 map 很可能会比 unordered_map 跑得快）。

代码如下：

/*
    Program: P3806.cpp
    Author: 1l6suj7
    DateTime: 2024-01-26 20:14:36
    Description:
*/

#include <bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define lp(i, j, n) for(int i = j; i <= n; ++i)
#define dlp(i, n, j) for(int i = n; i >= j; --i)
#define mst(n, v) memset(n, v, sizeof(n))
#define mcy(n, v) memcpy(n, v, sizeof(v))
#define INF 1e18
#define MAX4 0x3f3f3f3f
#define MAX8 0x3f3f3f3f3f3f3f3f
#define pii pair<int, int>
#define pll pair<ll, ll>
#define co(x) cerr << (x) << ' '
#define cod(x) cerr << (x) << endl
#define fi first
#define se second
#define eps 1e-8
#define lc(x) ((x) << 1)
#define rc(x) ((x) << 1 ^ 1)
#define pb(x) emplace_back(x)

using namespace std;

const int N = 10010;

struct edge { int v, nxt, w; } E[N << 1];
int en, hd[N];

void add(int u, int v, int w) { E[++en] = { v, hd[u], w }, hd[u] = en; }

int n, m, qs[N], ans[N];

int sz[N], dep[N], son[N], dfc, L[N], R[N], nd[N];
map<int, int> cnt;
void dfs1(int u, int fa) {
    sz[u] = 1, L[u] = R[u] = ++dfc, nd[dfc] = u;
    for(int i = hd[u]; i; i = E[i].nxt) {
        int v = E[i].v;
        if(v == fa) continue;
        dep[v] = dep[u] + E[i].w, dfs1(v, u), sz[u] += sz[v], R[u] = max(R[u], R[v]);
        if(sz[v] > sz[son[u]]) son[u] = v;
    }
}

void calc(int grd, int u) {
    lp(i, 1, m) {
        int t = dep[grd] + qs[i] - dep[u] + dep[grd];
        ans[i] |= cnt.count(t);
    }
}

void add(int u) { ++cnt[dep[u]]; }

void del(int u) { int t = --cnt[dep[u]]; if(!t) cnt.erase(dep[u]); }

void dfs2(int u, int fa, int kp) {
    for(int i = hd[u]; i; i = E[i].nxt) {
        int v = E[i].v;
        if(v == fa || v == son[u]) continue;
        dfs2(v, u, 0);
    }
    if(son[u]) dfs2(son[u], u, 1);
    for(int i = hd[u]; i; i = E[i].nxt) {
        int v = E[i].v;
        if(v == fa || v == son[u]) continue;
        lp(j, L[v], R[v]) calc(u, nd[j]);
        lp(j, L[v], R[v]) add(nd[j]);
    } calc(u, u), add(u);
    if(kp) return;
    lp(i, L[u], R[u]) del(nd[i]);
}

signed main() {
    // freopen("P3806.in", "r", stdin);
    // freopen("P3806.out", "w", stdout);
#ifndef READ
    ios::sync_with_stdio(false);
    cin.tie(0);
#endif
    cin >> n >> m;
    int u, v, w;
    lp(i, 1, n - 1) cin >> u >> v >> w, add(u, v, w), add(v, u, w);
    lp(i, 1, m) cin >> qs[i];
    dfs1(1, 0), dfs2(1, 0, 0);
    lp(i, 1, m) cout << (ans[i] ? "AYE" : "NAY") << endl;
    return 0;
}

例题 3：P2634 [国家集训队] 聪聪可可

很板，就不放代码了。

例题 4：[IOI2011] Race

也很板。

题意：一棵结点数为 n n n 的树，求其长度为 k k k 的简单路径上边的数量的最小值，若不存在输出 -1。

可以维护到根节点距离为 d d d 的结点到根节点边的数量的最小值，计算答案与合并显然，不过由于加点时要取 min，所以加点前要做个备份，删点时直接倒着扫一遍备份数组即可。

代码如下：

/*
    Program: P4149.cpp
    Author: 1l6suj7
    DateTime: 2024-01-26 21:14:54
    Description:
*/

#include <bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define lp(i, j, n) for(int i = j; i <= n; ++i)
#define dlp(i, n, j) for(int i = n; i >= j; --i)
#define mst(n, v) memset(n, v, sizeof(n))
#define mcy(n, v) memcpy(n, v, sizeof(v))
#define INF 1e18
#define MAX4 0x3f3f3f3f
#define MAX8 0x3f3f3f3f3f3f3f3f
#define pii pair<int, int>
#define pll pair<ll, ll>
#define co(x) cerr << (x) << ' '
#define cod(x) cerr << (x) << endl
#define fi first
#define se second
#define eps 1e-8
#define lc(x) ((x) << 1)
#define rc(x) ((x) << 1 ^ 1)
#define pb(x) emplace_back(x)

using namespace std;

const int N = 200010, M = 1000010;

int n, K, ans = MAX4;

struct edge { int v, nxt; ll w; } E[N << 1];
int en, hd[N];

void add(int u, int v, ll w) { E[++en] = { v, hd[u], w }, hd[u] = en; }

int sz[N], son[N], dfc, L[N], R[N], nd[N], cnt[N]; ll dep[N];
map<ll, int> res;
void dfs1(int u, int fa) {
    sz[u] = 1, L[u] = R[u] = ++dfc, nd[dfc] = u;
    for(int i = hd[u]; i; i = E[i].nxt) {
        int v = E[i].v;
        if(v == fa) continue;
        dep[v] = dep[u] + E[i].w, cnt[v] = cnt[u] + 1, dfs1(v, u), sz[v] += sz[u], R[u] = max(R[u], R[v]);
        if(sz[v] > sz[son[u]]) son[u] = v;
    }
}

void calc(int grd, int u) {
    int t = dep[grd] + K - dep[u] + dep[grd];
    if(res.count(t)) ans = min(ans, cnt[u] - cnt[grd] + res[t] - cnt[grd]);
}

pll bck[N];
int bn;
void add(int u) {
    int & t = res[dep[u]];
    if(!t || cnt[u] < t) bck[++bn] = { dep[u], t }, t = cnt[u];
}

void dfs2(int u, int fa, int kp, int pre) {
    for(int i = hd[u]; i; i = E[i].nxt) {
        int v = E[i].v;
        if(v == fa || v == son[u]) continue;
        dfs2(v, u, 0, bn + 1);
    }
    if(son[u]) dfs2(son[u], u, 1, bn);
    for(int i = hd[u]; i; i = E[i].nxt) {
        int v = E[i].v;
        if(v == fa || v == son[u]) continue;
        lp(j, L[v], R[v]) calc(u, nd[j]);
        lp(j, L[v], R[v]) add(nd[j]);
    } calc(u, u), add(u);
    if(kp) return;
    dlp(i, bn, pre) {
        int & t = res[bck[i].fi]; t = bck[i].se;
        if(!t) res.erase(bck[i].fi);
    }
    bn = 0;
}

signed main() {
    // freopen("P4149.in", "r", stdin);
    // freopen(".out", "w", stdout);
#ifndef READ
    ios::sync_with_stdio(false);
    cin.tie(0);
#endif
    cin >> n >> K;
    int u, v; ll w;
    lp(i, 1, n - 1) cin >> u >> v >> w, ++u, ++v, add(u, v, w), add(v, u, w);
    dep[1] = 1, dfs1(1, 0), dfs2(1, 0, 0, 1);
    if(ans ^ MAX4) cout << ans << endl;
    else cout << -1 << endl;
    return 0;
}

其他题目

[CF600E] Lomsat gelral

[CF1009F] Dominant Indices

[CF375D] Tree and Queries

1l6suj7
关注
  • 27
    点赞
  • 踩
  • 22
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
专栏目录
dsu:终端机中的每日站立笔记
03-31
DSU 终端中的每日站立笔记。 一个简单的Golang工具，用于基于Unix的系统，可以轻松添加和管理我的日常站立笔记。 昨天我做了什么，今天我要做什么。如何使用git clone ...
超微H12DSU-IN用户手册
03-24
超微H12DSU-IN用户手册
超微H11DSU-IN用户手册
03-24
超微H11DSU-IN用户手册
dsu_setup_Ver2.2_Build_17051810.zip
04-09
moxa设备搜索工具DSU
ACS800 DSU 硬件手册
11-07
ACS800 二极管供电单元固件手册 ，，，，，，
FME学习之旅---day27
summer_corner的博客
05-17 199
我们付出一些成本，时间的或者其他，最终总能收获一些什么。
使用多实例学习进行乳腺癌组织病理学图像分类和定位
qq_47896523的博客
05-16 1103
乳腺癌是女性最主要的死亡原因，病理学家根据病理切片中观察到的各种视觉特征（例如细胞核的形态特征、细胞核的微观和宏观结构等）做出决定。计算机辅助诊断（CAD）系统可以帮助病理学家自动做出决策。卷积神经网络是最广泛使用的深度学习框架，用于学习图像类别之间的复杂判别特征。多年来，VGG16 [3] 和 ResNet18 [4] 等各种 CNN 架构在海量 ImageNet 数据集上产生了出色的结果。CNN 被用于医学图像以产生最先进的结果。为图像中存在的所有类别提供了定位信息。
WEB转Flutter基础学习笔记（内含vue和flutter对比）
小易不止于搬砖的博客
05-17 675
笔者是一名web前端，记录在转栈flutter时的学习笔记，内涵和vue的对比，能够辅助前端同学更容易理解flutter
Page对象的学习
2301_77523055的博客
05-14 1005
ASP.NET允许开发者创建自定义控件和用户控件来扩展页面的功能。自定义控件是从现有的ASP.NET控件派生出来的新控件，而用户控件则是将多个控件组合成一个可重用的单元。这些控件可以像其他ASP.NET控件一样在页面上使用，并通过Page对象进行交互。Page_Init主要用于在控件加载之前进行初始化设置和添加事件处理程序。Page_Load主要用于处理页面回发和加载数据，以及执行其他需要在页面加载时完成的逻辑。由于Page_Init在Page_Load之前触发，因此在Page_Init。
学习kong官方文档
好好学习日记
05-17 275
当然，我会用更简洁的方式重新解释这两个概念：Helm和Kubectl是Kubernetes生态系统中常用的两个工具，它们各自有不同的用途：
推荐系统学习笔记（一）
weixin_44880995的博客
05-15 149
同类策略（精排中的两种模型）天然互斥，并且（两条召回通道）效果会相互影响，避免干扰。反转：有的指标立刻影响，有的需要长期观测-------尽快推全也可以长期观测。10%作为holdout桶，实验使用剩余的90%，做diff（需要归一化）不同策略（添加召回通道，优化粗排模型）通常不会干扰，可以作为正交的两层。保留10%，完全不受实验影响，可以考察整个部门对业务指标的贡献。同层互斥----避免一个用户被两个实验影响。不同层正交----每层独立随机分配用户。推全：新层，与其他层正交，90%用户。
DB类的学习
2301_77523055的博客
05-15 367
"不返回结果集" 意味着执行 SQL 语句后，你不会得到一个可以逐行读取的数据集。相反，你可能会得到一个表示受影响的行数的整数，或者只是一个表示命令是否成功执行的确认。
RustGUI学习（iced/iced_aw）之扩展小部件（十九）：如何使用context_menu部件来创建右击菜单？
用沸腾的热血，支付我们的人生吧！
05-15 73
本专栏是学习Rust的GUI库iced的合集，将介绍iced涉及的各个小部件分别介绍，最后会汇总为一个总的程序。iced是RustGUI中比较强大的一个，目前处于发展中（即版本可能会改变），本专栏基于版本0.12.1.
docker学习和常用命令
OutRoading的博客
05-14 185
镜像仓库datahub。
ptrade从零开始学习量化交易第3期【如何写一个简单但是完整的策略】
GJ57017_wx的博客
05-17 135
开始写策略 简单但是完整的策略 先来看一个简单但是完整的策略: def initialize(context): set_universe('600570.SS') def handle_data(context, data): pass 一个完整策略只需要两步: set_universe: 设置我们要操作的股票池，上面的例子中，只操作一支股票: '600570.SS'，恒生电子。所有的操作只能对股票池的标的进行。 实现一个函数: handle_data。 这是一个完整的策略，但
【从零开始学习Redis | 第十一篇】快速介绍Redis持久化策略
fckbb的博客
05-14 1019
Redis 作为一种快速、高效的内存数据库，被广泛应用于缓存、消息队列、会话存储等场景。然而，由于其特性是基于内存的，一旦服务器进程退出，内存中的数据就会丢失。为了解决这一问题，Redis 提供了持久化功能，可以将内存中的数据保存到硬盘上，以便在服务器重启时恢复数据。持久化是 Redis 的关键特性之一，对于确保数据的安全性和可靠性至关重要。本文将深入探讨 Redis 的持久化策略，包括常用的持久化方式以及各自的优缺点。
Android OpenMAX（八）如何学习OMXNodeInstance
青山渺渺
05-15 237
OMXNodeInstance学习技巧
【AI学习】简单聊聊后训练（Post-Training）的重要性
最新发布
bylander的博客
05-17 633
模型生成的输出质量比网上的大多数内容都要高。因此，让模型自己思考似乎更有道理，而不仅仅是训练来模仿网络上的内容。所以，我认为从第一性原理上来说，这是有说服力的。我会说，我们通过后训练取得了很多进步。
线性回归学习笔记
julyclj55555的博客
05-15 268
换句话说，多元线性回归可以有足够的精确性，但对于该精确性的合理解释会比较难，因为在多个属性中，可能有其中多组子属性的组合都能拟合到近似最优的模型效果，但对不同组的子属性的组合解释，会因为属性不同而有很大区别，这就是统计学家莱奥·布雷曼口中的“罗生门”（Rashomon）。它的值越大代表，拟合的结果越可信；将“罗生门”深挖一步，就是机器学习和统计学在认识论上的差异：统计学讲究的是“知其然，知其所以然”，它不仅要找出数据之间的关联性，还要挖出背后的因果性，给计算出的结果赋予令人信服的解释才是统计的核心。
moxa dsu串口虚拟软件
08-30
Moxa DSU是一款串口虚拟软件，主要用于在计算机上创建虚拟串口，方便用户与串口设备进行通信。 通过Moxa DSU软件，用户可以模拟多个串口设备，使其在计算机上以虚拟串口的形式存在。这样，用户可以通过计算机与虚拟...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值