「ZJOI2008」树的统计-树链剖分

给定一颗$n$个节点的树,节点编号为$1$到$n$,每个节点都有一个权值$w_i$。

有以下三种操作或询问:

I. $CHANGE u t$ : 把结点$u$的权值改为$t$

II. $QMAX u v$: 询问从点$u$到点$v$的路径上的节点的最大权值

III. $QSUM u v$: 询问从点$u$到点$v$的路径上的节点的权值和

链接

Luogu P2590

题解

树链剖分模版题。

进行树链剖分之后,由于注意到这里只有单点修改,所以我们用一个$zkw$线段树来维护这些信息。

  • 对于$CHANGE$操作,我们直接更改叶子节点的值,然后进行$zkw$线段树的从下到上的更新操作,直到根。

  • 对于$QMAX$操作,我们就用树链剖分的常规操作,每一步求一个最大值,然后更新答案即可。$QSUM$同理。

事实上,这个题凸显出来$zkw$线段树在单点修改上真是不一般的快,以我的垃圾coding水平居然能在bzoj上排到第一页。

$zkw$大法好!

代码

点击切换显示状态
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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
#include <cstdio>
#include <vector>
#include <cctype>
using namespace std;
const int MAXN = 31000;

inline int max(int a,int b){
return b>a?b:a;
}

namespace fast_io {
inline char read() {
//return getchar();
static const int IN_LEN = 1000000;
static char buf[IN_LEN], *s, *t;
return s==t?(((t=(s=buf)+fread(buf,1,IN_LEN,stdin))== s)?-1:*s++) : *s++;
}
inline void read(int &x) {
static bool iosig;
static char c;
for (iosig = false, c = read(); !isdigit(c); c = read()) {
if (c == '-') iosig = true;
if (c == -1) return;
}
for (x = 0; isdigit(c); c = read())
x = ((x + (x << 2)) << 1)+(c^48);
if (iosig) x = -x;
}
inline void read(char *a){
static char c = read();
while(c!= -1&&(c == ' '||c =='\n'))
c = read();
while(c!= -1&&c!=' '&&c!='\n')
*a++ = c,c = read();
*a = 0;
}
const int OUT_LEN = 1000000;
char obuf[OUT_LEN], *ooh = obuf;
inline void print(char c) {
if(ooh == obuf+OUT_LEN) fwrite(obuf,1,OUT_LEN,stdout),ooh=obuf;
*ooh++ = c;
}
inline void print(int x) {
static int buf[30], cnt;
if(x == 0)
print('0');
else {
if (x < 0) print('-'), x = -x;
for (cnt = 0; x; x /= 10) buf[++cnt] = x % 10 + 48;
while (cnt) print((char)buf[cnt--]);
}
}
inline void print(char *a){
while(*a) print(*a++);
}
inline void flush() {
fwrite(obuf, 1, ooh - obuf, stdout);
}
}using namespace fast_io;

int maxn[MAXN<<2],sumn[MAXN<<2];
int n,M,q,num[MAXN];

//top数组是链子最上面的节点,fa是父节点,son是重儿子节点,siz是子树大小,dep是dfs深度
int top[MAXN],fa[MAXN],son[MAXN],siz[MAXN],dep[MAXN];
//id_to 是i号id指向的节点编号,id就是i号节点的dfs编号
int id_to[MAXN],id[MAXN];
int cnt = 1;

vector<int> edge[MAXN];

//zkw线段树初始化
inline void init_seg(){
for(M = 1;M<n+2;M<<=1);
for(int i = 1;i<=n;i++)
maxn[M+i] = sumn[M+i] = num[id_to[i]];
for(int i = M-1;i;i--)
maxn[i]=max(maxn[i<<1],maxn[i<<1|1]),sumn[i]=sumn[i<<1]+sumn[i<<1|1];
}

//zkw线段树区间最值
inline int max_seg(int l,int r){
int ans = -100000;
for(l=l+M-1,r=r+M+1;l^r^1;l>>=1,r>>=1){
if(~l&1 && maxn[l^1]>ans) ans = maxn[l^1];
if(r&1 && maxn[r^1]>ans) ans = maxn[r^1];
}
return ans;
}

//zkw线段树区间求和
inline int sum_seg(int l,int r){
int ans = 0;
for(l=l+M-1,r=r+M+1;l^r^1;l>>=1,r>>=1){
if(~l&1) ans += sumn[l^1];
if(r&1) ans += sumn[r^1];
}
return ans;
}

//zkw线段树修改
inline void upd_seg(int pos,int x){
int nown = pos+M;
maxn[nown] = sumn[nown] = x;
while(nown>>=1){
maxn[nown] = max(maxn[nown<<1],maxn[nown<<1|1]);
sumn[nown] = sumn[nown<<1]+sumn[nown<<1|1];
}
}

//树链剖分的第一个dfs
inline void dfs1(int nown,int f,int depth){
son[nown] = 0,dep[nown] = depth,siz[nown] = 1,fa[nown] = f;
int maxsum = 0;
for(int i = 0;i<edge[nown].size();i++){
int to = edge[nown][i];
if(to == f) continue;
dfs1(to,nown,depth+1);
siz[nown]+=siz[to];
if(siz[to]>maxsum) maxsum = siz[to],son[nown] = to;
}
}

//树链剖分的第二个dfs
inline void dfs2(int nown,int topf){
id[nown] = cnt;
id_to[cnt++] = nown;
top[nown] = topf;
if(!son[nown]) return;
dfs2(son[nown],topf);
for(int i = 0;i<edge[nown].size();i++){
int to = edge[nown][i];
if(to == son[nown]||to == fa[nown]) continue;
dfs2(to,to);
}
}

//查询路径和
inline int query_sum(int x,int y){
int ans = 0;
while(top[x]!=top[y]){
if(dep[top[x]]<dep[top[y]]) swap(x,y);
ans+=sum_seg(id[top[x]],id[x]);
x = fa[top[x]];
}
if(dep[x]>dep[y]) swap(x,y);
ans+=sum_seg(id[x],id[y]);
return ans;
}

//查询路径最大值
inline int query_max(int x,int y){
int ans = -100000;
while(top[x]!=top[y]){
if(dep[top[x]]<dep[top[y]]) swap(x,y);
ans = max(ans,max_seg(id[top[x]],id[x]));
x = fa[top[x]];
}
if(dep[x]>dep[y]) swap(x,y);
ans = max(ans,max_seg(id[x],id[y]));
return ans;
}

//修改某个节点
inline void update(int pos,int x){
upd_seg(id[pos],x);
}

//初始化图和线段树还有两次dfs
void init(){
read(n);
int a,b;
for(int i = 1;i<=n-1;i++){
read(a),read(b);
edge[a].push_back(b);
edge[b].push_back(a);
}
for(int i = 1;i<=n;i++)
read(num[i]);
dfs1(1,0,1);
dfs2(1,1);
init_seg();
}

//进行操作以及回答询问
void solve(){
read(q);
char op[20];
int a,b;
for(int i = 1;i<=q;i++){
read(op),read(a),read(b);
if(op[1] == 'M')
print(query_max(a,b)),print('\n');
else if(op[1] == 'S')
print(query_sum(a,b)),print('\n');
else if(op[1] == 'H')
update(a,b);
}
}

int main(){
init();
solve();
flush();
return 0;
}