首页 > 其他分享 >树链剖分学习笔记

树链剖分学习笔记

时间:2023-01-31 21:34:35浏览次数:62  
标签:剖分 int top 笔记 树链 dep 节点 son id

怕到时候忘了,来写一篇笔记

前置芝士:树的存储与遍历,\(dfs\) 序,线段树。

树链剖分的思想及能解决的问题:

树链剖分用于将树分割成若干条链的形式,以维护树上路径的信息。

具体来说,将整棵树剖分为若干条链,使它组合成线性结构,然后用其他的数据结构维护信息。

树链剖分(树剖/链剖)有多种形式,如 重链剖分,长链剖分 和用于 Link/cut Tree 的剖分(有时被称作“实链剖分”),大多数情况下(没有特别说明时),“树链剖分”都指“重链剖分”。

重链剖分可以将树上的任意一条路径划分成不超过 \(O(\log n)\) 条连续的链,每条链上的点深度互不相同(即是自底向上的一条链,链上所有点的 \(LCA\) 为链的一个端点)。

重链剖分还能保证划分出的每条链上的节点 DFS 序连续,因此可以方便地用一些维护序列的数据结构(如线段树)来维护树上路径的信息。

重链剖分

我们给出一些定义:

定义 重子节点 表示其子节点中子树最大的子结点。如果有多个子树最大的子结点,取其一。如果没有子节点,就无重子节点。

定义:轻子节点表示剩余的所有子结点。

从这个结点到重子节点的边为重边。

到其他轻子节点的边为轻边。

若干条首尾衔接的重边构成重链。

把落单的结点也当作重链,那么整棵树就被剖分成若干条重链。

以上来自Oi-wiki

树链剖分的实现

以洛谷的模板题为例。

先给出以下定义:

\(fa[u]\) 表示\(u\)节点的父节点。

\(dep[u]\) 表示\(u\)节点的深度。

\(top[u]\) 表示\(u\)节点所在链的顶端。

\(son[u]\) 表示\(u\)节点的重儿子\(v\)的编号。

\(siz[u]\) 表示以\(u\)为根的子树的大小。

\(id[u]\) 表示\(u\)在树链剖分的\(dfs\)序。

树剖的实现过程分为:

\(dfs1:\)

\(dfs1\)中我们要处理出\(fa,dep,siz,son\)数组。

代码实现如下:

void dfs1(int u, int fath)  //处理fa,dep,siz,son
{
	fa[u] = fath;
	dep[u] = dep[fath] + 1;
	siz[u] = 1;
	for(auto v:e[u])
	{
		if(v == fath) continue;
		dfs1(v, u);
		siz[u] += siz[v];
		if(siz[v] > siz[son[u]])
			son[u] = v;
	}
}

\(dfs2:\)

\(dfs2\)中我们遵循优先走重边的原则,处理出我们的\(dfs\)序,也就是处理出\(top,id\)数组。

代码实现如下:

void dfs2(int u, int fst)  //处理top,id,fst表示当前链的顶端
{
	id[u] = ++ idx;
	a[idx] = w[u]; //a数组表示节点u的权值在线段树中的位置权值
	top[u] = fst;
	if(!son[u]) return; //没有重儿子意味着没有子节点
	dfs2(son[u], fst);  //跟重儿子一条链
	for(auto v:e[u])
	{
		if(v == fa[u] || v == son[u]) continue; //轻儿子既不是父节点也不是重儿子
		dfs2(v, v);  //以自己为链顶重新开一条链
	}
}

树剖求解\(LCA\)(最近公共祖先):

思路:不断的将两个节点中所在链的链顶深度深的节点往上跳,知道两个节点所在链相同,即\(top[u] == top[v]\),此时两个节点中深度较低的节点即为\(LCA\)。

代码如下:

int lca(int u, int v)
{
	while(top[u] != top[v])
	{
		if(dep[top[u]] < dep[top[v]]) swap(u, v);
		u = fa[top[u]];
	}
	if(dep[u] > dep[v]) swap(u, v);
	return u; 
}

操作一:将树从 \(x\) 到结点 \(y\) 最短路径上所有节点的值都加上\(z\)

这个时候就要祭出我们的线段树了,在\(dfs2\)中我们采取了优先走重儿子的原则,所以在一条链中的\(dfs\)序一定是连续的,我们能够根据这一特点快速的在线段树中修改某一段的权值。

思路与求解\(LCA\)基本相同,代码:

void trlist_update(int u, int v, int k)
{
	while(top[u] != top[v])
	{
		if(dep[top[u]] <= dep[top[v]]) swap(u, v);
		update(id[top[u]], id[u], k, 1, n, 1);
		u = fa[top[u]];
	}
	if(dep[u] > dep[v]) swap(u, v);
	update(id[u], id[v], k, 1, n, 1);
}

操作二:求树从 \(x\) 到 \(y\) 结点最短路径上所有节点的值之和

理解了\(LCA\)和树上修改权值后,查询其实也不难了,只要将操作一中的修改操作改为查询操作即可,代码如下:

int trlist_query(int u, int v)
{
	int res = 0;
	while(top[u] != top[v])
	{
		if(dep[top[u]] <= dep[top[v]]) swap(u, v);
		res += query(id[top[u]], id[u], 1, n, 1);
		u = fa[top[u]];
	}
	if(dep[u] > dep[v]) swap(u, v);
	res += query(id[u], id[v], 1, n, 1);
	return res;
}

操作三:将以 \(x\) 为根节点的子树内所有节点值都加上 \(z\)

因为我们在对树进行剖分的过程中使用的是 \(dfs\) ,所以每一颗子树内的 \(dfs\) 序也是连续的,利用我们之前处理出来的 \(siz\) 数组就能轻松解决这个问题。

代码如下:

void son_update(int u, int k)
{
	update(id[u], id[u] + siz[u] - 1, k, 1, n, 1);
}

操作四:求以 \(x\) 为根节点的子树内所有节点值之和

跟操作三基本一样,这里不多赘述,直接放代码:

int son_query(int u)
{
	return query(id[u], id[u] + siz[u] - 1, 1, n, 1);
}

完整代码

因为笔者很懒就直接放以前的AC代码了,码风稍有不同,加上了\(long long\)和取模等题目要求。

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=1e5+5;
int n,m,r,p,op,idx,cnt;
int head[N],dep[N],top[N],son[N],siz[N],fa[N],w[N],a[N],id[N];
struct edge
{
	int to,next;
}e[N<<1];
struct segment_tree
{
	int l,r,val,laz_tag;
}t[N<<2];

inline void push_up(int x)
{
	t[x].val=(t[x<<1].val+t[x<<1|1].val)%p;
}

inline void push_down(int x)
{
	if(!t[x].laz_tag) return;
	
	t[x<<1].laz_tag+=t[x].laz_tag;
	t[x<<1|1].laz_tag+=t[x].laz_tag;
	
	t[x<<1].val+=(t[x<<1].r-t[x<<1].l+1)*t[x].laz_tag;
	t[x<<1].val%=p;
	t[x<<1|1].val+=(t[x<<1|1].r-t[x<<1|1].l+1)*t[x].laz_tag;
	t[x<<1|1].val%=p;
	
	t[x].laz_tag=0;
}

inline void build(int l,int r,int x)
{
	t[x].l=l;
	t[x].r=r;
	if(l==r)
	{
		t[x].val=a[l]%p;
		return;
	}
	int mid=l+r>>1;
	build(l,mid,x<<1);
	build(mid+1,r,x<<1|1);
	push_up(x);
}

inline void update(int nl,int nr,int k,int l,int r,int x)
{
	if(nl<=l&&r<=nr)
	{
		t[x].val+=(r-l+1)*k;
		t[x].laz_tag+=k;
		return;
	}
	push_down(x);
	int mid=l+r>>1;
	if(nl<=mid) update(nl,nr,k,l,mid,x<<1);
	if(nr>mid) update(nl,nr,k,mid+1,r,x<<1|1);
	push_up(x);
}

inline int query(int nl,int nr,int l,int r,int x)
{
	if(nl<=l&&r<=nr)
	{
		return t[x].val;
	}
	push_down(x);
	int mid=l+r>>1,res=0;
	if(nl<=mid) res+=query(nl,nr,l,mid,x<<1);
	if(nr>mid) res+=query(nl,nr,mid+1,r,x<<1|1);
	return res;
}

inline void add_edge(int u,int v)
{
	e[++cnt].to=v;
	e[cnt].next=head[u];
	head[u]=cnt;
}

inline void dfs1(int u,int fath)
{
	dep[u]=dep[fath]+1;
	fa[u]=fath;
	siz[u]=1;
	for(int i=head[u];i;i=e[i].next)
	{
		int v=e[i].to;
		if(v==fath) continue;
		dfs1(v,u);
		siz[u]+=siz[v];
		if(siz[v]>siz[son[u]])
		  son[u]=v;
	}
}

inline void dfs2(int u,int fst)
{
	id[u]=++idx;
	a[idx]=w[u];
	top[u]=fst;
	if(!son[u]) return;
	dfs2(son[u],fst);
	for(int i=head[u];i;i=e[i].next)
	{
		int v=e[i].to;
		if(v==fa[u]||v==son[u]) continue;
		dfs2(v,v);
	}
}

inline void trlist_update(int u,int v,int w)
{
	w%=p;
	while(top[u]!=top[v])
	{
		if(dep[top[u]]<dep[top[v]]) swap(u,v);
		update(id[top[u]],id[u],w,1,n,1);
		u=fa[top[u]];
	}
	if(dep[u]>dep[v]) swap(u,v);
	update(id[u],id[v],w,1,n,1);
}

inline int trlist_query(int u,int v)
{
	int res=0;
	while(top[u]!=top[v])
	{
		if(dep[top[u]]<dep[top[v]]) swap(u,v);
		res+=query(id[top[u]],id[u],1,n,1);
		res%=p;
		u=fa[top[u]];
	}
	if(dep[u]>dep[v]) swap(u,v);
	res+=query(id[u],id[v],1,n,1);
	res%=p;
	return res;
}

inline void son_update(int u,int w)
{
	update(id[u],id[u]+siz[u]-1,w,1,n,1);
}

inline int son_query(int u)
{
	return query(id[u],id[u]+siz[u]-1,1,n,1)%p;
}

signed main()
{
	std::ios::sync_with_stdio(false);
	std::cin.tie(NULL);
	std::cout.tie(NULL);
	cin>>n>>m>>r>>p;
	for(int i=1;i<=n;i++)
	  cin>>w[i];
	for(int i=1;i<n;i++)
	{
		int u,v;
		cin>>u>>v;
		add_edge(u,v);
		add_edge(v,u);
	}
	dfs1(r,0);
	dfs2(r,r);
	build(1,n,1);
	while(m--)
	{
		cin>>op;
		if(op==1)
		{
			int u,v,w;
			cin>>u>>v>>w;
			trlist_update(u,v,w);
		}
		if(op==2)
		{
			int u,v;
			cin>>u>>v;
			cout<<trlist_query(u,v)<<'\n';
		}
		if(op==3)
		{
			int u,w;
			cin>>u>>w;
			son_update(u,w);
		}
		if(op==4)
		{
			int u;
			cin>>u;
			cout<<son_query(u)<<'\n';
		}
	}
	return 0;
}

到这里就能很轻松的解决树上修改等一系列问题了,接下来给出几道例题。

\(1\).P3038 [USACO11DEC]Grass Planting G(思考如何边权转点权)

\(2\). P4427 [BJOI2018]求和(预处理+树剖)

\(3\). P1505 [国家集训队]旅游(很裸的树剖,不懂为什么评紫,注意节点编号是从\(0\)开始的即可)

标签:剖分,int,top,笔记,树链,dep,节点,son,id
From: https://www.cnblogs.com/syz1552672065/p/17080811.html

相关文章