Description
有一棵點數爲 N 的樹,以點 1 爲根,且樹點有邊權。然後有 M 個
操作,分爲三種:
操作 1 :把某個節點 x 的點權增加 a 。
操作 2 :把某個節點 x 爲根的子樹中所有點的點權都增加 a 。
操作 3 :詢問某個節點 x 到根的路徑中所有點的點權和。
Input
第一行包含兩個整數 N, M 。表示點數和操作數。接下來一行 N 個整數,表示樹中節點的初始權值。接下來 N-1
行每行三個正整數 fr, to , 表示該樹中存在一條邊 (fr, to) 。再接下來 M 行,每行分別表示一次操作。其中
第一個數表示該操作的種類( 1-3 ) ,之後接這個操作的參數( x 或者 x a ) 。
Output
對於每個詢問操作,輸出該詢問的答案。答案之間用換行隔開。
Sample Input
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
Sample Output
6
9
13
HINT
對於 100% 的數據, N,M<=100000 ,且所有輸入數據的絕對值都不會超過 10^6 。
Solution
這題本來是樹鏈剖分裸題。但是這題可以用線段樹的
代碼:
#include<cstdio>
#include<algorithm>
using namespace std;
template<typename T>inline void read(T &x){
T f=1;char ch=getchar();
for(;ch<'0'||ch>'9';ch=getchar())if(ch=='-')f=-1;
for(x=0;ch>='0'&&ch<='9';ch=getchar())x=x*10+ch-'0';
x*=f;
}
typedef long long LL;
const int maxn=100010;
int n,m,num,head[maxn],dfn[maxn],cnt,tL[maxn],tR[maxn],dep[maxn];
LL a[maxn],w[maxn];
struct edge{
int to,next;
}e[maxn<<1];
struct Segment_Tree{
#define lc x<<1
#define rc x<<1|1
int L[maxn<<2],R[maxn<<2];
LL add[maxn<<2],tag[maxn<<2];
void Build(int x,int l,int r){
L[x]=l;R[x]=r;
if(l==r)return add[x]=tag[x]=0,void();
int mid=(l+r)>>1;
Build(lc,l,mid);
Build(rc,mid+1,r);
}
void pushdown(int x){
if(add[x]){
add[lc]+=add[x];
add[rc]+=add[x];
add[x]=0;
}
if(tag[x]){
tag[lc]+=tag[x];
tag[rc]+=tag[x];
tag[x]=0;
}
}
void Add1(int x,int l,int r,LL val){
if(R[x]<l||L[x]>r)return;
if(L[x]>=l&&R[x]<=r)return add[x]+=val,void();
pushdown(x);
Add1(lc,l,r,val);
Add1(rc,l,r,val);
}
void Add2(int x,int l,int r,LL val){
if(R[x]<l||L[x]>r)return;
if(L[x]>=l&&R[x]<=r)return tag[x]+=val,void();
pushdown(x);
Add2(lc,l,r,val);
Add2(rc,l,r,val);
}
LL Query(int x,int pos){
if(L[x]==R[x])return add[x]+tag[x]*dep[dfn[pos]];
pushdown(x);
int mid=(L[x]+R[x])>>1;
if(pos<=mid)return Query(lc,pos);
else return Query(rc,pos);
}
}tree;
void add(int u,int v){
e[++num].to=v;e[num].next=head[u];head[u]=num;
e[++num].to=u;e[num].next=head[v];head[v]=num;
}
void Dfs(int x,int fa){
dep[dfn[tL[x]=++cnt]=x]=dep[fa]+1;
w[x]=w[fa]+a[x];
for(int i=head[x];i;i=e[i].next)
if(e[i].to!=fa)Dfs(e[i].to,x);
tR[x]=cnt;
}
int main(){
read(n);read(m);
for(int i=1;i<=n;i++)read(a[i]);
for(int i=1,u,v;i<n;i++)
read(u),read(v),add(u,v);
Dfs(1,0);tree.Build(1,1,n);
for(int i=1;i<=n;i++)tree.Add1(1,tL[i],tL[i],w[i]);
while(m--){
int opt,x;LL val;
read(opt);read(x);
if(opt==1){
read(val);
tree.Add1(1,tL[x],tR[x],val);
}
else if(opt==2){
read(val);
tree.Add2(1,tL[x],tR[x],val);
tree.Add1(1,tL[x],tR[x],-val*(dep[x]-1));
}
else printf("%lld\n",tree.Query(1,tL[x]));
}
return 0;
}