#2897 「HAOI2015」树上操作
题面
有一棵点数为 N 的树,以点 1 为根,且树点有边权。然后有 M 个操作,分为三种:
1:把某个节点 x 的点权增加 a 。
2:把某个节点 x 为根的子树中所有点的点权都增加 a 。
3:询问某个节点 x 到根的路径中所有点的点权和。
输入
第一行包含两个整数 N,M。表示点数和操作数。
接下来一行 N 个整数,表示树中节点的初始权值。
接下来 N−1 行每行两个正整数 fr,to , 表示该树中存在一条边 (fr,to)
再接下来 M 行,每行分别表示一次操作。其中第一个数表示该操作的种类(1-3) ,之后接这个操作的参数(x 或者 x a) 。
输出
对于每个询问操作,输出该询问的答案。答案之间用换行隔开。
样例输入
5 5
1 2 3 4 5
1 2
1 4
2 3
2 5
3 3
1 2 1
3 5
2 1 2
3 3
样例输出
6
9
13
提示
对于 100%的数据,
N
,
M
≤
100000
N,M≤100000
N,M≤100000,且所有输入数据的绝对值都不会超过
1
0
6
10^6
106 。
SOL
树链剖分的模板题。
按轻重链剖分之后,转成一般的区间问题,然后用线段树乱搞。
对于1、2两个操作,可以用一个change函数解决,都转化成区间修改的操作来处理。询问就按正常的方法写。
代码:
#include<bits/stdc++.h>
#define N 100005
#define int long long
using namespace std;
inline int rd(){
int data=0,w=1;static char ch=0;
while(!isdigit(ch)&&ch!='-')ch=getchar();
if(ch=='-')w=-1,ch=getchar();
while(isdigit(ch))data=(data<<1)+(data<<3)+ch-'0',ch=getchar();
return data*w;
}
int n,m,ans,ins;
int cnt,first[N];
struct node{int v,nxt;}e[N<<1];
inline void add(int u,int v){e[++cnt].v=v;e[cnt].nxt=first[u];first[u]=cnt;}
int w[N],W[N],seg[N<<1],siz[N],son[N],f[N],dep[N],top[N];
void dfs1(int u,int fa,int d){
siz[u]=1;f[u]=fa;dep[u]=d;
for(int register i=first[u];i;i=e[i].nxt){
int register v=e[i].v;
if(v==fa)continue;
dfs1(v,u,d+1);
siz[u]+=siz[v];if(siz[son[u]]<siz[v])son[u]=v;
}
}
void dfs2(int u,int tp){
seg[u]=++seg[0];
w[seg[0]]=W[u];
top[u]=tp;
if(!son[u])return;
dfs2(son[u],tp);
for(int register i=first[u];i;i=e[i].nxt){
int register v=e[i].v;
if(v==f[u]||v==son[u])continue;
dfs2(v,v);
}
}
#define lc (p<<1)
#define rc (p<<1|1)
struct segment_tree{int sum,add;}t[N<<2];
inline void pushup(int p){t[p].sum=t[lc].sum+t[rc].sum;}
inline void pushdown(int p,int l){
if(!t[p].add)return;
t[lc].add+=t[p].add;t[rc].add+=t[p].add;
t[lc].sum+=t[p].add*(l-(l>>1));t[rc].sum+=t[p].add*(l>>1);
t[p].add=0;
}
void build(int p,int l,int r){
if(l==r){t[p].sum=w[l];return;}
int register mid=l+r>>1;
build(lc,l,mid);
build(rc,mid+1,r);
pushup(p);
}
void change(int p,int l,int r,int v,int ql,int qr){
if(ql<=l&&qr>=r){t[p].sum+=v*(r-l+1);t[p].add+=v;return;}
int register mid=l+r>>1;
pushdown(p,r-l+1);
if(ql<=mid)change(lc,l,mid,v,ql,qr);
if(qr>mid)change(rc,mid+1,r,v,ql,qr);
pushup(p);
}
void query(int p,int l,int r,int ql,int qr){
if(ql<=l&&qr>=r){ans+=t[p].sum;return;}
int mid=l+r>>1;
pushdown(p,r-l+1);
if(ql<=mid)query(lc,l,mid,ql,qr);
if(qr>mid)query(rc,mid+1,r,ql,qr);
}
void ret(int x,int y){
while(f[top[x]]!=f[top[y]]){
if(dep[top[x]]<dep[top[y]])swap(x,y);
query(1,1,n,seg[top[x]],seg[x]);
x=f[top[x]];
}
if(dep[x]>dep[y])swap(x,y);
query(1,1,n,seg[x],seg[y]);
}
signed main(){
n=rd();m=rd();
for(int register i=1;i<=n;i++)W[i]=rd();
for(int register i=1;i<n;i++){int register fr=rd(),to=rd();add(fr,to);add(to,fr);}
dfs1(1,0,1);dfs2(1,1);build(1,1,n);
while(m--){ins=rd();
if(ins==1){int register x=rd(),v=rd();change(1,1,n,v,seg[x],seg[x]);}
if(ins==2){int register x=rd(),v=rd();change(1,1,n,v,seg[x],seg[x]+siz[x]-1);}
if(ins==3){int register x=rd();ans=0;ret(1,x);printf("%lld\n",ans);}
}
return 0;
}