「HAOI2015」树上操作

赤裸裸的树剖啊。

传送门

洛谷P3178

BZOJ4034

题解

裸的树链剖分,没啥好说的。直接搞就好了

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
#include<cstdio>
using namespace std;
typedef long long LL;
const int maxn=100005;
int n,m,tot,lnk[maxn],son[maxn*2],w[maxn],nxt[maxn*2],idx,siz[maxn],wson[maxn],fa[maxn],top[maxn],seg[maxn],rev[maxn];LL ans;
inline int read()
{
int ret=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-f;ch=getchar();}
while(ch>='0'&&ch<='9'){ret=ret*10+ch-'0';ch=getchar();}
return ret*f;
}
struct SegmentTree
{
struct Node{LL Sum,Tag;}Tree[maxn*4];
void PushUp(int rt){Tree[rt].Sum=Tree[rt*2].Sum+Tree[rt*2+1].Sum;}
void PushDown(int rt,int LC,int RC)
{
if(!Tree[rt].Tag) return;
Tree[rt*2].Sum+=Tree[rt].Tag*LC;
Tree[rt*2+1].Sum+=Tree[rt].Tag*RC;
Tree[rt*2].Tag+=Tree[rt].Tag;
Tree[rt*2+1].Tag+=Tree[rt].Tag;
Tree[rt].Tag=0;
}
void Build(int L=1,int R=n,int rt=1)
{
if(L==R){Tree[rt].Sum=w[rev[L]];return;}
int M=(L+R)>>1;
Build(L,M,rt*2);
Build(M+1,R,rt*2+1);
PushUp(rt);
}
void RangeUpdate(int LL,int RR,int delta,int L=1,int R=n,int rt=1)
{
if(LL<=L&&R<=RR)
{
Tree[rt].Sum+=(long long)delta*(R-L+1);
Tree[rt].Tag+=delta;
return;
}
int M=(L+R)/2;
PushDown(rt,M-L+1,R-M);
if(LL<=M) RangeUpdate(LL,RR,delta,L,M,rt*2);
if(M<RR) RangeUpdate(LL,RR,delta,M+1,R,rt*2+1);
PushUp(rt);
}
LL QueryRange(int LL,int RR,int L=1,int R=n,int rt=1)
{
if(LL<=L&&R<=RR) return Tree[rt].Sum;
int M=(L+R)/2;::LL ret=0;
PushDown(rt,M-L+1,R-M);
if(LL<=M) ret+=QueryRange(LL,RR,L,M,rt*2);
if(M<RR) ret+=QueryRange(LL,RR,M+1,R,rt*2+1);
return ret;
}
}T;
inline void add_e(int x,int y){tot++;son[tot]=y;nxt[tot]=lnk[x];lnk[x]=tot;}
void DFS1(int now)
{
siz[now]=1;
for(int i=lnk[now];i;i=nxt[i])
{
if(son[i]==fa[now]) continue;
fa[son[i]]=now;
DFS1(son[i]);
siz[now]+=siz[son[i]];
}
for(int i=lnk[now];i;i=nxt[i])
if(son[i]!=fa[now]&&siz[son[i]]>siz[wson[now]])
wson[now]=son[i];
}
void DFS2(int now)
{
seg[now]=++idx;rev[seg[now]]=now;
if(now==1) top[now]=1;
if(wson[now])
{
top[wson[now]]=top[now];
DFS2(wson[now]);
}
for(int i=lnk[now];i;i=nxt[i])
{
if(son[i]==fa[now]||son[i]==wson[now]) continue;
top[son[i]]=son[i];
DFS2(son[i]);
}
}
int main()
{
n=read();m=read();
for(int i=1;i<=n;i++) w[i]=read();
for(int i=1;i<n;i++)
{
int a=read(),b=read();
add_e(a,b);add_e(b,a);
}
DFS1(1);DFS2(1);T.Build();
while(m--)
{
int opt=read(),x=read();
if(opt==1)
{
int a=read();
T.RangeUpdate(seg[x],seg[x],a);
}
else if(opt==2)
{
int a=read();
T.RangeUpdate(seg[x],seg[x]+siz[x]-1,a);
}
else
{
ans=0;
while(x>0)
{
ans+=T.QueryRange(seg[top[x]],seg[x]);
x=fa[top[x]];
}
printf("%lld\n",ans);
}
}
return 0;
}