【模板·樹剖】 洛谷 P3384 【模板】樹鏈剖分
阿新 • • 發佈:2018-12-24
題目:樹鏈剖分
注意:線段樹區間處理時左右區間不要顛倒。
程式碼:
#include<bits/stdc++.h>
using namespace std;
#define read(x) scanf("%d",&x)
#define maxn 100000
#define ll long long
struct Node {
int lstid;
int w,fa;
int sz,d,hson;
int tp;
};
int n,m;
int rt,md;
Node a[maxn+5];
vector<int> tr[maxn+5] ;
int cnt=0;
int frtid[maxn+5];
struct segTree {
#define lson (o*2)
#define rson (o*2+1)
#define mid (L+(R-L)/2)
int c[maxn*10+5],lzy[maxn*10+5];
int len[maxn*10+5];
void push_up(int o) {
c[o]=(c[lson]+c[rson])%md;
}
void build(int o,int L,int R) {
len[o]=R-L+1;
if(L==R) {
c[o]=a[ frtid[L]].w;
return ;
}
build(lson,L,mid),build(rson,mid+1,R);
push_up(o);
}
int P,Q;
void push_down(int o) {
lzy[lson]=((ll)lzy[lson]+lzy[o])%md,lzy[rson]=((ll)lzy[rson]+lzy[o])%md;
c[lson]=((ll)c[lson]+(ll)lzy[o]*len[lson]%md)%md,c[rson]=((ll)c[rson]+(ll)lzy[o]*len[rson]%md)%md;
lzy[o]=0;
}
int Qry(int o,int L,int R) {
if(L>Q||R<P) return 0;
if(L>=P&&R<=Q) return c[o];
push_down(o);
int ans=(Qry(lson,L,mid)+Qry(rson,mid+1,R))%md;
return ans;
}
int Query(int x,int y) {
if(x>y) swap(x,y);
P=x,Q=y;
return Qry(1,1,n);
}
void Upd(int o,int L,int R,int d) {
if(L>Q||R<P) return ;
if(L>=P&&R<=Q) {
c[o]=((ll)c[o]+(ll)d*len[o])%md;
lzy[o]=(d+lzy[o])%md;
return ;
}
push_down(o);
Upd(lson,L,mid,d),Upd(rson,mid+1,R,d);
push_up(o);
}
void Update(int x,int y,int d) {
if(x>y) swap(x,y);
P=x,Q=y;
Upd(1,1,n,d);
}
};
segTree sgtr;
struct HLD{
void update(int x,int y,int z) {
while(a[x].tp!=a[y].tp) {
if(a[a[x].tp].d<a[a[y].tp].d) swap(x,y);
sgtr.Update(a[x].lstid,a[a[x].tp].lstid,z);
x=a[a[x].tp].fa;
}
sgtr.Update(a[x].lstid,a[y].lstid,z);
}
int query(int x,int y) {
int ans=0;
while(a[x].tp!=a[y].tp) {
if(a[a[x].tp].d<a[a[y].tp].d) swap(x,y);
ans=(ans+ sgtr.Query(a[x].lstid,a[a[x].tp].lstid) )%md;
x=a[a[x].tp].fa;
}
ans=(ans+ sgtr.Query(a[x].lstid,a[y].lstid) )%md;
return ans;
}
};
HLD hld;
void readin() {
read(n),read(m),read(rt),read(md);
for(int i=1;i<=n;i++) read(a[i].w);
for(int i=1;i<n;i++) {
int x,y;
read(x),read(y);
tr[x].push_back(y);
tr[y].push_back(x);
}
}
void dfs1(int x,int fa) {
a[x].fa=fa;
a[x].sz=1,a[x].d=a[fa].d+1;
int maxson=0,maxsz=-1;
for(int i=0;i<tr[x].size();i++) {
int y=tr[x][i];
if(y==fa) continue;
dfs1(y,x);
a[x].sz+=a[y].sz;
if(a[y].sz>maxsz) {
maxsz=a[y].sz;
maxson=y;
}
}
a[x].hson=maxson;
}
void dfs2(int x,int tp,int fa) {
a[x].tp=tp;
a[x].lstid=++cnt,frtid[cnt]=x;
if(a[x].hson==0) return;
dfs2(a[x].hson,tp,x);
for(int i=0;i<tr[x].size();i++) {
int y=tr[x][i];
if(y==fa||y==a[x].hson) continue;
dfs2(y,y,x);
}
}
int main() {
readin();
dfs1(rt,0);
dfs2(rt,rt,0);
sgtr.build(1,1,n);
while(m--) {
int opr;
read(opr);
if(opr==1) {
int x,y,z;
read(x),read(y),read(z);
hld.update(x,y,z);
} else if(opr==2) {
int x,y;
read(x),read(y);
printf("%d\n",hld.query(x,y)%md);
} else if(opr==3) {
int x,z;
read(x),read(z);
sgtr.Update(a[x].lstid,a[x].lstid+a[x].sz-1,z);
} else {
int x;
read(x);
printf("%d\n",sgtr.Query(a[x].lstid,a[x].lstid+a[x].sz-1)%md);
}
}
return 0;
}