树链剖分 模板 洛谷P3384

题目要求：

已知一棵包含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为根节点的子树内所有节点值之和

代码

#include<bits/stdc++.h>
using namespace std;
#define int long long
int n,root,num_e,num_s,mod,m,opt,u,v;
inline void _read(int &x){
x=0;char c=getchar();
while(c<'0'||c>'9') c=getchar();
while(c>='0'&&c<='9'){
x=(x<<1)+(x<<3)+c-'0';
c=getchar();
}
}
inline void read(int &x){
_read(x);
}
inline void read(int &x,int &y){
_read(x);
_read(y);
}
inline void read(int &x,int &y,int &z){
_read(x);
_read(y);
_read(z);
}
inline void read(int &x,int &y,int &z,int &o){
_read(x);
_read(y);
_read(z);
_read(o);
}
#define maxn 1000005
struct edge{
int to,nex;
}e[maxn];
int head[maxn],mp[maxn],a[maxn];
void add(int x,int y){
e[++num_e].to=y;e[num_e].nex=head[x];head[x]=num_e;
e[++num_e].to=x;e[num_e].nex=head[y];head[y]=num_e;
}
struct tre{
int siz,dep,fa,top,dis,wson,s,e;
}t[maxn];
struct point{
int l,r,lazy,sum;
}p[maxn*6];
void dfs1(int,int);
void dfs2(int,int);
void build_tree(int,int,int);
void push_down(int);
void up_date(int,int,int,int);
int ask(int,int,int);
void solve1();
void solve2();
void solve3();
void solve4();
#undef int
int main(){
read(n,m,root,mod);
for(int i=1;i<=n;i++) read(a[i]);
u,v;
for(int i=1;i<n;i++) read(u,v),add(u,v);//cout<<"   !!";
dfs1(root,1);//for(int i=1;i<=n;i++) cout<<t[i].wson<<' '; cout<<endl;
dfs2(root,root);
build_tree(1,num_s,1);
for(int i=1;i<=m;i++){
read(opt);
if(opt==1) solve1();
else if(opt==2) solve2();
else if(opt==3) solve3();
else solve4();
}

return 0;
}
#define int long long
void dfs1(int x,int depth){
t[x].siz=1;
t[x].dep=depth;
for(int i=head[x];i;i=e[i].nex){
int y=e[i].to;
if(t[x].fa!=y){
t[y].fa=x;
dfs1(y,depth+1);
t[x].siz+=t[y].siz;
if(t[t[x].wson].siz<t[y].siz) t[x].wson=y;
}
}
}
void dfs2(int x,int fir){
t[x].top=fir;//cout<<"  **"<<x<<' '<<t[x].wson;
t[x].s=++num_s;
mp[num_s]=x;
if(t[x].wson){
dfs2(t[x].wson,fir);
for(int i=head[x];i;i=e[i].nex){
int y=e[i].to;
if(y!=t[x].fa&&y!=t[x].wson){
dfs2(y,y);
}
}
}
t[x].e=num_s;
}
/***********************************
愉快的树剖到此结束。。
5 5 2 24
7 3 7 8 0
1 2
1 5
3 1
4 1
3 4 2
3 2 2
4 5
1 5 1 3
2 1 3
痛苦的线段树已经开始。。。
************************************/
void build_tree(int l,int r,int num){
p[num].l=l,p[num].r=r;p[num].lazy;
if(l==r){
p[num].sum=a[mp[l]];
return;
}
int m=(l+r)>>1;
build_tree(l,m,num<<1);
build_tree(m+1,r,num<<1|1);
p[num].sum=p[num<<1].sum+p[num<<1|1].sum;
}
inline void push_down(int x){
if(!p[x].lazy) return;
p[x<<1].sum+=(p[x<<1].r-p[x<<1].l+1)*p[x].lazy; p[x<<1].lazy+=p[x].lazy;
p[x<<1|1].sum+=(p[x<<1|1].r-p[x<<1|1].l+1)*p[x].lazy; p[x<<1|1].lazy+=p[x].lazy;
p[x].lazy=0;
}
void up_date(int l,int r,int x,int num){
if(l<=p[num].l&&r>=p[num].r){
p[num].lazy+=x;
p[num].sum+=x*(p[num].r-p[num].l+1);
return;
}
push_down(num);
int m=(p[num].l+p[num].r)>>1;
if(l<=m) up_date(l,r,x,num<<1);
if(r>m) up_date(l,r,x,num<<1|1);
p[num].sum=p[num<<1].sum+p[num<<1|1].sum;
}
int ask(int l,int r,int num){
if(l<=p[num].l&&r>=p[num].r) return p[num].sum;
int m=(p[num].l+p[num].r)>>1;int summ=0;
push_down(num);
if(l<=m) summ+=ask(l,r,num<<1);
if(r>m) summ+=ask(l,r,num<<1|1);
return summ;
}
void solve1(){
int x,y,w;
read(x,y,w);
int f1=t[x].top,f2=t[y].top;
while(f1!=f2){
if(t[f1].dep<t[f2].dep){
swap(f1,f2);
swap(x,y);
}
up_date(t[f1].s,t[x].s,w,1);
x=t[f1].fa;
f1=t[x].top;
}
if(t[x].dep<t[y].dep) up_date(t[x].s,t[y].s,w,1);
else up_date(t[y].s,t[x].s,w,1);
}
void solve2(){
int x,y;
read(x,y);
int f1=t[x].top,f2=t[y].top,summ=0;
while(f1!=f2){
if(t[f1].dep<t[f2].dep){
swap(f1,f2);
swap(x,y);
}
summ+=ask(t[f1].s,t[x].s,1);
summ%=mod;
x=t[f1].fa;
f1=t[x].top;
}
if(t[x].dep<t[y].dep) summ+=ask(t[x].s,t[y].s,1);
else summ+=ask(t[y].s,t[x].s,1);
printf("%lld\n",summ%mod);
}
void solve3(){
int x,w;
read(x,w);//cout<<"    !! "<<t[x].s<<' '<<t[x].e<<endl;
up_date(t[x].s,t[x].e,w,1);
}
void solve4(){
int x;
read(x);
printf("%lld\n",ask(t[x].s,t[x].e,1)%mod);
}
#undef int