树链剖分的学习,我是从树链剖分-1和树链剖分-2来学会的。
模板题
简单讲一下树链剖分吧,就是它首先第一遍dfs1,就将f[i] ,dep[i] ,sz[i] ,son[i]都预处理好,它们分别代表节点i的父节点,深度,子节点的数量,重节点;第二遍dfs2,就将id[i],rk[i],top[i],分别代表节点i的dfs序,dfs序为i的节点,节点i所在链的顶端节点。如果你有lca和线段树的知识,它跟lca的倍增的原理差不多,但是它每次是跳该节点的top[i]节点。然后用到线段树,根据它规定的搜素顺序(从重链开始到轻链)来建树,很简单,普通线段树怎么写就怎么写,关键是你线段树维护的是dfs序1-n每个节点对应的信息,然后它找的时候边跳边用线段树去维护。
#include<bits/stdc++.h>
#define ll long long
#define m (l+r)/2
#define ls o*2
#define rs o*2 +1
using namespace std;
const int maxn = 2e5+10;
int f[maxn],d[maxn],sz[maxn],son[maxn],top[maxn],id[maxn],rk[maxn],cnt;
vector<int>G[maxn];
ll tr[maxn*4] ,lz[maxn*4];
int n,mod,a[maxn];
void dfs1(int u,int fa,int dep)
{
f[u] = fa;
d[u] = dep;
sz[u] = 1;
for(auto v:G[u])
{
if(v==fa) continue;
dfs1(v,u,dep+1);
sz[u] += sz[v];
if(sz[v]>sz[son[u]]||!son[u])
son[u] = v;
}
}
void dfs2(int u,int t)
{
top[u] = t;
id[u] = ++cnt;
rk[cnt] = a[u];
if(!son[u]) return ;
dfs2(son[u],t);
for(auto v:G[u])
{
if(v!=son[u]&&v!=f[u])
dfs2(v,v);
}
}
void build(int o,int l,int r)
{
if(l==r)
{
tr[o] = rk[l]%mod;
return ;
}
build(ls,l,m);
build(rs,m+1,r);
tr[o] = (tr[ls]+tr[rs])%mod;
}
void push(int o,int l,int r)
{
if(lz[o]!=0){
tr[ls] = (tr[ls]%mod + 1LL*(m-l+1)*lz[o]%mod)%mod ;
tr[rs] = (tr[rs] %mod +1LL*(r-m)*lz[o]%mod)%mod;
lz[ls] = (lz[o]+lz[ls])%mod , lz[rs] = (lz[rs]+lz[o])%mod;
lz[o] = 0;
}
}
void up(int o,int l,int r,int ql,int qr,int z)
{
if(ql<=l&&qr>=r)
{
tr[o] = (tr[o]+1LL*(r-l+1)*z%mod)%mod ;
lz[o] = (lz[o]+z%mod)%mod;
return ;
}
push(o,l,r);
if(ql<=m) up(ls,l,m,ql,qr,z);
if(qr>m) up(rs,m+1,r,ql,qr,z);
tr[o] = (tr[ls] + tr[rs])%mod;
}
ll qu(int o,int l,int r,int ql,int qr)
{
if(ql<=l&&qr>=r)
return tr[o]%mod;
push(o,l,r);
int ans = 0;
if(ql<=m) ans =(ans+qu(ls,l,m,ql,qr))%mod;
if(qr>m) ans = (ans+qu(rs,m+1,r,ql,qr))%mod;
return ans%mod ;
}
void up_path(int x,int y,int z)
{
int fx = top[x] , fy =top[y];
while(fx!=fy)
{
if(d[fx]>d[fy]){
up(1,1,cnt,id[fx],id[x],z);
x = f[fx];
}
else {
up(1,1,cnt,id[fy],id[y],z);
y = f[fy];
}
fx = top[x] , fy = top[y];
}
if(id[x]<=id[y]) up(1,1,cnt,id[x],id[y],z);
else up(1,1,cnt,id[y],id[x],z);
}
ll qu_path(int x,int y)
{
ll ans = 0;
int fx = top[x] , fy = top[y];
while(fx!=fy)
{
if(d[fx]>d[fy])
{
ans = (ans + qu(1,1,cnt,id[fx],id[x]))%mod;
x = f[fx];
}
else {
ans = (ans+qu(1,1,cnt,id[fy],id[y]))%mod;
y = f[fy];
}
fx = top[x] , fy = top[y];
}
if(id[x]<=id[y]) ans = (ans+qu(1,1,cnt,id[x],id[y]))%mod;
else ans = (ans+qu(1,1,cnt,id[y],id[x]))%mod;
return ans%mod ;
}
int main()
{
int u,v;
int q,rt;scanf("%d%d%d%d",&n,&q,&rt,&mod);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
for(int i=1;i<n;i++)
{
scanf("%d%d",&u,&v);
G[u].push_back(v);
G[v].push_back(u);
}
dfs1(rt,0,1);
dfs2(rt,rt);
build(1,1,n);
int op,x,y,z;
for(int i=1;i<=q;i++)
{
scanf("%d",&op);
if(op==1){
scanf("%d%d%d",&x,&y,&z);
up_path(x,y,z%mod);
}
if(op==2){
scanf("%d%d",&x,&y);
printf("%lld\n",qu_path(x,y));
}
if(op==3)
{
scanf("%d%d",&x,&z);
up(1,1,cnt,id[x],id[x]+sz[x]-1,z%mod);
}
if(op==4)
{
scanf("%d",&x);
printf("%lld\n",qu(1,1,cnt,id[x],id[x]+sz[x]-1));
}
}
}