【bzoj4034】树上操作 树链剖分+线段树

AC通道：http://www.lydsy.com/JudgeOnline/problem.php?id=4034

【题解】

用树剖来做的话，做法很显然，比模板题还简单。

不过，据说可以用dfs序搞一搞。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<ctime>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long ll;
#define FILE "read"
#define MAXN 100010
#define up(i,j,n) for(ll i=j;i<=n;++i)
#define dn(i,j,n) for(ll i=j;i>=n;--i)
#define cmax(a,b) a=max(a,b)
#define cmin(a,b) a=min(a,b)
namespace INIT{
char buf[1<<15],*fs,*ft;
inline char getc(){return (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<15,stdin),fs==ft))?0:*fs++;}
inline ll read(){
ll x=0,f=1; char ch=getc();
while(!isdigit(ch)) {if(ch=='-') f=-1; ch=getc();}
while(isdigit(ch)) {x=x*10+ch-'0'; ch=getc();}
return x*f;
}
}using namespace INIT;
struct node{ll y,next;}e[MAXN*2];
ll n,m,len,z,a[MAXN],Link[MAXN],size[MAXN],f[MAXN],w[MAXN],deep[MAXN],son[MAXN],top[MAXN],tr[MAXN*8],delta[MAXN*8];
void insert(ll x,ll y){e[++len].next=Link[x];Link[x]=len;e[len].y=y;}
void dfs(ll x){
size[x]=1; son[x]=0;
for(ll i=Link[x];i;i=e[i].next){
if(e[i].y==f[x]) continue;
f[e[i].y]=x; deep[e[i].y]=deep[x]+1;
dfs(e[i].y); size[x]+=size[e[i].y];
if(size[e[i].y]>size[son[x]]) son[x]=e[i].y;
}
}
void dfs(ll x,ll temp){
w[x]=++z; top[x]=temp;
if(son[x]) dfs(son[x],temp);
for(ll i=Link[x];i;i=e[i].next)
if(e[i].y!=f[x]&&e[i].y!=son[x]) dfs(e[i].y,e[i].y);
}
void pushdown(ll x,ll l,ll r){
ll delt=delta[x],mid=(l+r)>>1;
tr[x<<1]+=delt*(mid-l+1); tr[x<<1|1]+=delt*(r-mid);
delta[x<<1]+=delt; delta[x<<1|1]+=delt;
delta[x]=0;
}
void add(ll p,ll l,ll r,ll x,ll y,ll v){
pushdown(p,l,r);
if(x>r||y<l) return;
if(x<=l&&y>=r) {tr[p]+=v*(r-l+1); delta[p]+=v; return;}
ll mid=(l+r)>>1;
add(p<<1,l,mid,x,y,v); add(p<<1|1,mid+1,r,x,y,v);
tr[p]=tr[p<<1]+tr[p<<1|1];
}
ll find(ll p,ll l,ll r,ll x,ll y){
pushdown(p,l,r);
if(x>r||y<l) return 0;
if(x<=l&&y>=r) return tr[p];
ll mid=(l+r)>>1;
return find(p<<1,l,mid,x,y)+find(p<<1|1,mid+1,r,x,y);
}
ll solve(ll x){
ll ans=0;
while(x){
ans+=find(1,1,z,w[top[x]],w[x]);
x=f[top[x]];
}
return ans;
}
int main(){
freopen(FILE".in","r",stdin);
freopen(FILE".out","w",stdout);
n=read(); m=read();
up(i,1,n) a[i]=read();
up(i,1,n-1) {ll x=read(),y=read(); insert(x,y); insert(y,x);}
dfs(1); dfs(1,1);
up(i,1,n) add(1,1,z,w[i],w[i],a[i]);
up(i,1,m){
ll opt=read();
if(opt==1) {ll x=read(),v=read(); add(1,1,z,w[x],w[x],v);}
else if(opt==2) {ll x=read(),v=read(); add(1,1,z,w[x],w[x]+size[x]-1,v);}
else {ll x=read(); printf("%lld\n",solve(x));}
}
return 0;
}