博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
树链剖分入门
阅读量:5051 次
发布时间:2019-06-12

本文共 4942 字,大约阅读时间需要 16 分钟。

简介

树链剖分就是将树划分为多条链,将每条链映射到序列上,然后使用线段树,平衡树等数据结构来维护每条链的信息。

树剖将树链映射到序列上后用线段树等数据结构来维护树链信息,所以可以像区间修改,区间查询一样进行树上路径的修改、查询等操作

基本概念

重儿子:子树结点数目最多的儿子(size最大的点);

重边:父亲结点和重儿子连成的边;

重链:由多条重边连接而成的路径;

轻儿子: 除了重儿子,其余都为轻儿子;

轻边:重边之外的边;

1598496-20190803101812591-1644538026.png

红圈表示重儿子;

黑边表示重边;

由黑边连成的链即为重链

实现

第一遍dfs处理出重儿子,深度,父亲等信息

第二遍dfs处理出结点所在重链的链顶,dfs序

上图中的树处理完毕后dfs序如下

1598496-20190803101918245-1710963686.png

可以发现,由于我们优先dfs重儿子,所以重儿子结点的编号是连续的,于是一条重链就被映射成了一段连续的区间;

这样树上两点间的路径就被分割成了多个连续的区间,如

(12,8) 可分割成 12,2 - 6, 1 – 4, 8 四段

(11,13)可分成 2-6-11,1 – 4 -9 -13 两段

于是我们就可以使用线段树来进行树链修改与查询了

时间复杂度

每次都将路径分割成多个区间,区间操作可以在O(logn)内解决, 但如果分割成的区间数很多怎么办?!

那我们就来看下两点间的路径最多会分割成多少段。

这时候重儿子就发挥作用了,由于每条链的链顶都是一个轻儿子,轻儿子的大小肯定小于重儿子, 所以size[轻儿子]<=size[父亲]/2

这样从上往下每进入一条新链,结点的个数就会除2,所以经过的链数就是log级别的了。

树剖每次将路径分成log个区间,然后区间操作一般都会用到线段树之类的数据结构来维护,所以一般情况下一次操作的时间复杂度为(logn)^2

例题

树的统计

Description

一棵树上有n个节点,编号分别为1到n,每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成一些操作:

I. CHANGE u t : 把结点u的权值改为t

II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 I

II. QSUM u v: 询问从点u到点v的路径上的节点的权值和

注意:从点u到点v的路径上的节点包括u和v本身

Input

输入的第一行为一个整数n,表示节点的个数。接下来n – 1行,每行2个整数a和b,表示节点a和节点b之间有一条边相连。接下来n行,每行一个整数,第i行的整数wi表示节点i的权值。接下来1行,为一个整数q,表示操作的总数。接下来q行,每行一个操作,以“CHANGE u t”或者“QMAX u v”或者“QSUM u v”的形式给出。

对于100%的数据,保证1<=n<=30000,0<=q<=200000;中途操作中保证每个节点的权值w在-30000到30000之间。

Output

对于每个“QMAX”或者“QSUM”的操作,每行输出一个整数表示要求输出的结果。

Sample Input

41 22 34 14 2 1 312QMAX 3 4QMAX 3 3QMAX 3 2QMAX 2 3QSUM 3 4QSUM 2 1CHANGE 1 5QMAX 3 4CHANGE 3 6QMAX 3 4QMAX 2 4QSUM 3 4

Sample Output

412210656516

题解

树剖模板题,树链剖分后用线段树维护即可

#include 
#define lson (o << 1)#define rson (o << 1 | 1)using namespace std;const int N = 3e4 + 10;typedef long long ll;vector
G[N];const ll inf = 1e9;int n;int val[N];int fa[N];int son[N];int sze[N];int dep[N];void dfs1(int u, int f) { sze[u] = 1; fa[u] = f; son[u] = 0; dep[u] = dep[f] + 1; for (int i = 0; i < G[u].size(); i++) { int v = G[u][i]; if (v == f) continue; dfs1(v, u); sze[u] += sze[v]; if (sze[v] > sze[son[u]]) son[u] = v; }}int top[N];int cnt;int pos[N];int a[N];void dfs2(int u, int f, int t) { top[u] = t; pos[u] = ++cnt; a[cnt] = val[u]; if (son[u]) dfs2(son[u], u, t); for (int i = 0; i < G[u].size(); i++) { int v = G[u][i]; if (v == f || v == son[u]) continue; dfs2(v, u, v); }}ll sumv[N << 2];ll maxv[N << 2];void pushup(int o) { sumv[o] = sumv[lson] + sumv[rson]; maxv[o] = max(maxv[lson], maxv[rson]);}void build(int o, int l, int r) { if (l == r) { sumv[o] = a[l]; maxv[o] = a[l]; return; } int mid = (l + r) >> 1; build(lson, l, mid); build(rson, mid + 1, r); pushup(o);}void update(int o, int l, int r, int pos, ll v) { if (l == r) { sumv[o] = v; maxv[o] = v; return; } int mid = (l + r) >> 1; if (pos <= mid) update(lson, l, mid, pos, v); else update(rson, mid + 1, r, pos, v); pushup(o);}ll querysum(int o, int l, int r, int ql, int qr) { if (ql <= l && r <= qr) { return sumv[o]; } ll ans = 0; int mid = (l + r) >> 1; if (ql <= mid) ans += querysum(lson, l, mid, ql, qr); if (qr > mid) ans += querysum(rson, mid + 1, r, ql, qr); return ans;}ll querymax(int o, int l, int r, int ql, int qr) { if (ql <= l && r <= qr) { return maxv[o]; } ll ans = -inf; int mid = (l + r) >> 1; if (ql <= mid) ans = max(ans, querymax(lson, l, mid, ql, qr)); if (qr > mid) ans = max(ans, querymax(rson, mid + 1, r, ql, qr)); return ans;}ll calcsum(int u, int v) { ll ans = 0; while (top[u] != top[v]) {//当不在同一条链上 if (dep[top[u]] < dep[top[v]]) swap(u, v);//每次深度较大的点向上走 ans += querysum(1, 1, n, pos[top[u]], pos[u]); u = fa[top[u]];//进入新的链 } if (dep[u] < dep[v]) swap(u, v);//进入同一条链再求一次 ans += querysum(1, 1, n, pos[v], pos[u]); return ans;}ll calcmax(int u, int v) { ll ans = -inf; while (top[u] != top[v]) { if (dep[top[u]] < dep[top[v]]) swap(u, v); ans = max(ans, querymax(1, 1, n, pos[top[u]], pos[u])); u = fa[top[u]]; } if (dep[u] < dep[v]) swap(u, v); ans = max(ans, querymax(1, 1, n, pos[v], pos[u])); return ans;}int main() { //freopen("in.txt", "r", stdin); //freopen("out.txt", "w", stdout); scanf("%d", &n); for (int i = 1; i < n; i++) { int u, v; scanf("%d%d", &u, &v); G[u].push_back(v); G[v].push_back(u); } for (int i = 1; i <= n; i++) scanf("%d", &val[i]); dep[0] = 0; dfs1(1, 0); cnt = 0; dfs2(1, 0, 1); build(1, 1, n); int m; scanf("%d", &m); char ch[10]; for (int i = 1; i <= m; i++) { scanf("%s", ch); int l, r, k; ll v; switch(ch[1]) { case 'M': scanf("%d%d", &l, &r); printf("%lld\n", calcmax(l, r)); break; case 'S': scanf("%d%d", &l, &r); printf("%lld\n", calcsum(l, r)); break; case 'H': scanf("%d%lld", &k, &v); update(1, 1, n, pos[k], v); break; } } return 0;}

转载于:https://www.cnblogs.com/artoriax/p/11293972.html

你可能感兴趣的文章
BZOJ 1200 木梳
查看>>
【Linux】【C语言】菜鸟学习日志(一) 一步一步学习在Linxu下测试程序的运行时间...
查看>>
hostname
查看>>
SpringBoot使用其他的Servlet容器
查看>>
关于cookie存取中文乱码问题
查看>>
mysql 多表管理修改
查看>>
group by order by
查看>>
Oracle学习之简单查询
查看>>
log4j配置
查看>>
linux 配置SAN存储-IPSAN
查看>>
java学习笔记之String类
查看>>
pymysql操作mysql
查看>>
Linux服务器删除乱码文件/文件夹的方法
查看>>
牛腩记账本core版本源码
查看>>
Word Break II
查看>>
UVA 11082 Matrix Decompressing 矩阵解压(最大流,经典)
查看>>
jdk从1.8降到jdk1.7失败
查看>>
一些关于IO流的问题
查看>>
mongo备份操作
查看>>
8 -- 深入使用Spring -- 3...1 Resource实现类InputStreamResource、ByteArrayResource
查看>>