「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;
}