树链剖分学习笔记

最新推荐文章于 2025-02-13 21:09:48 发布

原创最新推荐文章于 2025-02-13 21:09:48 发布 · 334 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#树链剖分

树链剖分专栏收录该内容

3 篇文章

订阅专栏

写在前面，算法笔记只是针对个人复习用，要想学习这个算法还是要看更详细的博客或视频讲解。

树链剖分

基本概念不讲，讲一下各个函数的作用，以及基本实现原理；

首先要知道树链剖分的作用就是把树上问题转化为线性问题；

线性问题众所周知就有区间问题，有区间问题就有线段树这种东西；

所以树链剖分就是先把树拆成一条条链（线性），然后用线段树维护区间就行；

怎么拆是最难的，我觉得搞懂dfs序是前提，因为就是根据dfs序进行拆分的；

这里不解释了；

dfs1 这个函数就是相当于预处理，预处理出树的每个结点的 deep（深度），fa（父亲），size（子树大小），son（重儿子编号）；

void dfs1(int p,int fat){
	size[p]=1,fa[p]=fat;
	int mx=-1;
	for(int i=head[p];~i;i=edge[i].nex){
		int q=edge[i].to;
		if(q!=fat){
			deep[q]=deep[p]+1;
			dfs1(q,p);
			size[p]+=size[q];
			if(size[q]>mx) son[p]=q,mx=size[q]; 
		}
	}
}

dfs2 这个函数就是dfs序进行拆分了，求出每个点的 id（新编号），w（新编号点的权值），top（每个结点所在链的顶端）

void dfs2(int p,int topp){
	id[p]=++tot,w[tot]=a[p],top[p]=topp;
	if(!son[p]) return;//没有重儿子，也就是叶子结点 
	dfs2(son[p],topp);//先处理重儿子 
	for(int i=head[p];~i;i=edge[i].nex){
		int q=edge[i].to;
		if(q==fa[p]||q==son[p]) continue;
		dfs2(q,q);
	}
}

然后就是最重要的询问和修改函数了，因为已经转化为线性了，所以只要求出要维护的区间的就行（然后用线段树维护这个区间），就可以了；怎么求出要维护的区间，就跟求LCA的思路差不多；

LL query_chain(int x,int y){
	LL ans=0;
	while(top[x]!=top[y]){
		if(deep[top[x]]<deep[top[y]]) swap(x,y);//保证x结点的顶点更深 
		(ans+=query(id[top[x]],id[x],1))%=p;
		x=fa[top[x]]; 
	}
	if(deep[x]<deep[y]) swap(x,y);
	(ans+=query(id[y],id[x],1))%=p;
	return ans;
}
LL query_tree(int x){
	LL ans=0;
	(ans+=query(id[x],id[x]+size[x]-1,1))%=p;
	return ans;
}

全部代码来自这个题：

洛谷·P3384 【模板】轻重链剖分

代码：

#include<bits/stdc++.h>
#define LL long long
#define pa pair<int,int>
#define ls k<<1
#define rs k<<1|1
#define inf 0x3f3f3f3f
using namespace std;
const int N=100100;
const int M=50100;
const LL mod=10007;
LL a[N],p,w[N];
int head[N],cnt,n,m,r;
int size[N],fa[N],son[N],deep[N];
int top[N],id[N],tot;
struct Node{
	int to,nex;
}edge[N*2];
struct node{
	int l,r;
	LL w,lz;
}tr[N*4];
void add(int p,int q){edge[cnt].to=q,edge[cnt].nex=head[p],head[p]=cnt++;}
void dfs1(int p,int fat){
	size[p]=1,fa[p]=fat;
	int mx=-1;
	for(int i=head[p];~i;i=edge[i].nex){
		int q=edge[i].to;
		if(q!=fat){
			deep[q]=deep[p]+1;
			dfs1(q,p);
			size[p]+=size[q];
			if(size[q]>mx) son[p]=q,mx=size[q]; 
		}
	}
}
void dfs2(int p,int topp){
	id[p]=++tot,w[tot]=a[p],top[p]=topp;
	if(!son[p]) return;//没有重儿子，也就是叶子结点 
	dfs2(son[p],topp);//先处理重儿子 
	for(int i=head[p];~i;i=edge[i].nex){
		int q=edge[i].to;
		if(q==fa[p]||q==son[p]) continue;
		dfs2(q,q);
	}
}
////
void pp(int k){
	tr[k].w=tr[ls].w+tr[rs].w;
	tr[k].w%=p;
}
void pd(int k){
	if(tr[k].lz){
		(tr[ls].lz+=tr[k].lz)%=p,(tr[rs].lz+=tr[k].lz)%=p;
		(tr[ls].w+=1ll*(tr[ls].r-tr[ls].l+1)*tr[k].lz)%=p;
		(tr[rs].w+=1ll*(tr[rs].r-tr[rs].l+1)*tr[k].lz)%=p;
	}
	tr[k].lz=0;
}
void build(int l,int r,int k){
	tr[k].lz=tr[k].w=0,tr[k].l=l,tr[k].r=r;
	if(l==r){
		tr[k].w=(w[l]%p);
		return;
	}
	int d=(l+r)>>1;
	build(l,d,ls);
	build(d+1,r,rs);
	pp(k);
}
void update(int l,int r,LL w,int k){
	if(tr[k].l>=l&&tr[k].r<=r){
		(tr[k].lz+=w)%=p;
		(tr[k].w+=1ll*(tr[k].r-tr[k].l+1)*w)%=p;
		return;
	}
	pd(k);
	int d=(tr[k].l+tr[k].r)>>1;
	if(l<=d) update(l,r,w,ls);
	if(r>d) update(l,r,w,rs);
	pp(k);
}
LL query(int l,int r,int k){
	LL ans=0;
	if(tr[k].l>=l&&tr[k].r<=r) return tr[k].w%p;
	pd(k);
	int d=(tr[k].l+tr[k].r)>>1;
	if(l<=d) (ans+=query(l,r,ls))%=p;
	if(r>d) (ans+=query(l,r,rs))%=p;
	return ans%p;
}
///
LL query_chain(int x,int y){
	LL ans=0;
	while(top[x]!=top[y]){
		if(deep[top[x]]<deep[top[y]]) swap(x,y);//保证x结点的顶点更深 
		(ans+=query(id[top[x]],id[x],1))%=p;
		x=fa[top[x]]; 
	}
	if(deep[x]<deep[y]) swap(x,y);
	(ans+=query(id[y],id[x],1))%=p;
	return ans;
}
LL query_tree(int x){
	LL ans=0;
	(ans+=query(id[x],id[x]+size[x]-1,1))%=p;
	return ans;
}
void update_chain(int x,int y,LL z){
	while(top[x]!=top[y]){
		if(deep[top[x]]<deep[top[y]]) swap(x,y);//保证x结点的顶点更深 
		update(id[top[x]],id[x],z%p,1);
		x=fa[top[x]]; 
	}
	if(deep[x]<deep[y]) swap(x,y);
	update(id[y],id[x],z%p,1);
}
void update_tree(int x,LL z){
	update(id[x],id[x]+size[x]-1,z%p,1);
} 
int main(){
	memset(head,-1,sizeof(head));
	scanf("%d%d%d%lld",&n,&m,&r,&p);
	for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
	for(int i=1;i<n;i++){
		int x,y;scanf("%d%d",&x,&y);
		add(x,y),add(y,x);
	}
	dfs1(r,0);
	dfs2(r,r);
	build(1,n,1);//建线段树 
	for(int i=1;i<=m;i++){
		int op;scanf("%d",&op);
		if(op==1){
			int x,y;LL z;scanf("%d%d%lld",&x,&y,&z);
			update_chain(x,y,z);
		}
		else if(op==2){
			int x,y;scanf("%d%d",&x,&y);
			printf("%lld\n",query_chain(x,y));
		}
		else if(op==3){
			int x;LL z;scanf("%d%lld",&x,&z);
			update_tree(x,z); 
		}
		else{
			int x;scanf("%d",&x);
			printf("%lld\n",query_tree(x));
		}
	}
	return 0;
}