我们知道对一列数进行区间或单点加减,乘除和区间求值等操作可以用线段树或树状数组
那么,如何对带权树上一条路径中的数进行这样的操作呢?
此时就用到了线段树的树上版——树链剖分
树链剖分的目的在于把树变成一条线段,以方便区间操作
显然,我们无法让每一条树上路径中所有数在这条线段中相邻,但又不能让它们太分散
于是,我们想寻找一个使区间操作均摊复杂度较小的树-线段映射方法
介绍概念:
- 重儿子:父亲节点的所有儿子中子树结点数目最多(size最大)的结点;
- 轻儿子:父亲节点中除了重儿子以外的儿子;
- 重边:父亲结点和重儿子连成的边;
- 轻边:父亲节点和轻儿子连成的边;
- 重链:由多条重边连接而成的路径;
- 轻链:由多条轻边连接而成的路径;
介绍数组:
fa[i]:点i的父亲
top[i]:点i所在重链的顶点
si[i]:以点i为根的子树的节点数
son[i]:点i的重儿子
dfn[i]:点i的dfs序
dep[i]:点i的深度

比如上面这幅图中,用黑线连接的结点都是重结点,其余均是轻结点,
2-11就是重链,2-5就是轻链,用红点标记的就是该结点所在重链的起点,也就是下文提到的top结点,
还有每条边的值其实是进行dfs时的执行序号。
(图和说明来自这位大佬的博客)
遍历时,我们使用先序遍历,并先遍历重儿子,再遍历轻儿子
于是,dfn重儿子=dfn父亲+1,即一条重链上从上到下dfn相邻且依次增大
我们按照dfn序将节点的权值放到数列上,然后就可以愉快地对它进行线段树操作了!
树链剖分的两个性质:
1,如果(u, v)是一条轻边,那么size(v) < size(u)/2;
2,从根结点到任意结点的路所经过的轻重链的个数必定都小于logn;
可以证明,树链剖分中一次区间操作的时间复杂度为O((log2n)^2);
模板题:【模板】重链剖分(luogu)
Description
题目描述
如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作:
操作1: 格式: 1 x y z 表示将树从x到y结点最短路径上所有节点的值都加上z
操作2: 格式: 2 x y 表示求树从x到y结点最短路径上所有节点的值之和
操作3: 格式: 3 x z 表示将以x为根节点的子树内所有节点值都加上z
操作4: 格式: 4 x 表示求以x为根节点的子树内所有节点值之和
输入格式
第一行包含4个正整数N、M、R、P,分别表示树的结点个数、操作个数、根节点序号和取模数(即所有的输出结果均对此取模)。
接下来一行包含N个非负整数,分别依次表示各个节点上初始的数值。
接下来N-1行每行包含两个整数x、y,表示点x和点y之间连有一条边(保证无环且连通)
接下来M行每行包含若干个正整数,每行表示一个操作,格式如下:
操作1: 1 x y z
操作2: 2 x y
操作3: 3 x z
操作4: 4 x
输出格式
输出包含若干行,分别依次表示每个操作2或操作4所得的结果(对P取模)
Code
#include <cstdio>
#include <cstdlib>
#include <vector>
#include <algorithm>
#define ll long long
using namespace std;
const int N=1e5+10;
ll d[N],P,z;
int re[N],n,m,opt,x,y,r,fa[N],top[N],si[N],son[N],dfn[N],tot,rt,sum,dep[N];
struct node
{
int l,r,lc,rc;
ll sum,delay;
}f[N*2];
vector <int> link[N];
void dfs(int u,int f)
{
fa[u]=f,si[u]=1,dep[u]=dep[f]+1;
int size=link[u].size();
for(int i=0;i<size;i++)
{
int v=link[u][i];
if(v==f) continue;
dfs(v,u),si[u]+=si[v];
if(son[u]==0 || si[son[u]]<si[v]) son[u]=v;
}
}
void Dfs(int u)
{
dfn[u]=++tot,re[tot]=u;
if(!son[u]) return ;
top[son[u]]=top[u],Dfs(son[u]);
int size=link[u].size();
for(int i=0;i<size;i++)
{
int v=link[u][i];
if(v==fa[u] || v==son[u]) continue;
top[v]=v,Dfs(v);
}
}
ll gel(int g)
{
return f[g].r-f[g].l+1;
}
void push_up(int g)
{
f[g].sum=f[f[g].lc].sum+f[f[g].rc].sum;
f[g].sum%=P;
}
void push_down(int g)
{
if(f[g].delay==0) return ;
int lc=f[g].lc,rc=f[g].rc;
f[lc].sum+=gel(lc)*f[g].delay;
f[lc].delay+=f[g].delay;
f[lc].delay%=P,f[lc].sum%=P;
f[rc].sum+=gel(rc)*f[g].delay;
f[rc].delay+=f[g].delay;
f[rc].delay%=P,f[rc].sum%=P;
f[g].delay=0;
}
void build(int &g,int l,int r)
{
g=++sum;
f[g].l=l,f[g].r=r;
if(l==r)
{
f[g].sum+=d[re[l]];
return ;
}
int mid=(l+r)>>1;
build(f[g].lc,l,mid);
build(f[g].rc,mid+1,r);
push_up(g);
}
void add(int g,int l,int r,ll k)
{
if(f[g].l>=l && f[g].r<=r)
{
f[g].sum+=gel(g)*k,f[g].delay+=k;
f[g].sum%=P,f[g].delay%=P;
return ;
}
push_down(g);
int mid=(f[g].l+f[g].r)>>1;
if(r<=mid) add(f[g].lc,l,r,k);
else if(l>mid) add(f[g].rc,l,r,k);
else add(f[g].lc,l,mid,k),add(f[g].rc,mid+1,r,k);
push_up(g);
}
ll get(int g,int l,int r)
{
if(f[g].l>=l && f[g].r<=r)
return f[g].sum%P;
push_down(g);
int mid=(f[g].l+f[g].r)>>1;
if(r<=mid) return get(f[g].lc,l,r)%P;
else if(l>mid) return get(f[g].rc,l,r)%P;
else return (get(f[g].lc,l,mid)+get(f[g].rc,mid+1,r))%P;
}
void Add(int x,int y,ll k)
{
int px=top[x],py=top[y];
while(px!=py)
if(dep[px]>=dep[py])
{
add(rt,dfn[px],dfn[x],k);
x=fa[px],px=top[x];
}
else
{
add(rt,dfn[py],dfn[y],k);
y=fa[py],py=top[y];
}
if(dfn[x]>dfn[y]) add(rt,dfn[y],dfn[x],k);
else add(rt,dfn[x],dfn[y],k);
}
ll Get(int x,int y)
{
ll ans=0;
int px=top[x],py=top[y];
while(px!=py)
if(dep[px]>=dep[py])
{
ans+=get(rt,dfn[px],dfn[x]),ans%=P;
x=fa[px],px=top[x];
}
else
{
ans+=get(rt,dfn[py],dfn[y]),ans%=P;
y=fa[py],py=top[y];
}
if(dfn[x]>dfn[y]) ans+=get(rt,dfn[y],dfn[x]),ans%=P;
else ans+=get(rt,dfn[x],dfn[y]),ans%=P;
return ans%P;
}
int main()
{
scanf("%d%d%d%lld",&n,&m,&r,&P);
for(int i=1;i<=n;i++)
scanf("%lld",&d[i]);
for(int i=1;i<n;i++)
{
int u,v;
scanf("%d%d",&u,&v);
link[u].push_back(v);
link[v].push_back(u);
}
dfs(r,0);
top[r]=r,Dfs(r);
build(rt,1,tot);
// 1 x y z 表示将树从x到y结点最短路径上所有节点的值都加上z
// 2 x y 表示求树从x到y结点最短路径上所有节点的值之和
// 3 x z 表示将以x为根节点的子树内所有节点值都加上z
// 4 x 表示求以x为根节点的子树内所有节点值之和
while(m--)
{
scanf("%d",&opt);
if(opt==1)
{
scanf("%d%d%lld",&x,&y,&z);
z%=P;
if(z==0) continue;
Add(x,y,z);
}
else if(opt==2)
{
scanf("%d%d",&x,&y);
printf("%lld\n",Get(x,y));
}
else if(opt==3)
{
scanf("%d%lld",&x,&z);
z%=P;
if(z==0) continue;
add(rt,dfn[x],dfn[x]+si[x]-1,z);
}
else if(opt==4)
{
scanf("%d",&x);
printf("%lld\n",get(rt,dfn[x],dfn[x]+si[x]-1));
}
}
return 0;
}
来源:https://www.cnblogs.com/hsez-cyx/p/12251001.html