1. 程式人生 > >【模板·樹剖】 洛谷 P3384 【模板】樹鏈剖分

【模板·樹剖】 洛谷 P3384 【模板】樹鏈剖分

題目:樹鏈剖分

注意:線段樹區間處理時左右區間不要顛倒。

程式碼:

#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; }