Time Limit: 10 Sec Memory Limit: 162 MB
Submit: 17411 Solved: 7109
Description
一棵树上有n个节点,编号分别为1到n,每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成
一些操作: I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 I
II. QSUM u v: 询问从点u到点v的路径上的节点的权值和 注意:从点u到点v的路径上的节点包括u和v本身
Input
输入的第一行为一个整数n,表示节点的个数。接下来n – 1行,每行2个整数a和b,表示节点a和节点b之间有
一条边相连。接下来n行,每行一个整数,第i行的整数wi表示节点i的权值。接下来1行,为一个整数q,表示操作
的总数。接下来q行,每行一个操作,以“CHANGE u t”或者“QMAX u v”或者“QSUM u v”的形式给出。
对于100%的数据,保证1<=n<=30000,0<=q<=200000;中途操作中保证每个节点的权值w在-30000到30000之间。
Output
对于每个“QMAX”或者“QSUM”的操作,每行输出一个整数表示要求输出的结果。
Sample Input
4
1 2
2 3
4 1
4 2 1 3
12
QMAX 3 4
QMAX 3 3
QMAX 3 2
QMAX 2 3
QSUM 3 4
QSUM 2 1
CHANGE 1 5
QMAX 3 4
CHANGE 3 6
QMAX 3 4
QMAX 2 4
QSUM 3 4
Sample Output
4
1
2
2
10
6
5
6
5
16
LCT水过去,现在才发现链剖好写好调,这才是我的第三道LCT,我太菜了,唉
#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
const int MAXN=300000+100;
struct Line{
int to,nxt;
}line[MAXN*2+10];
int head[MAXN],tail,n,from,to;
void add_line(int from,int to){tail++;line[tail].to=to;line[tail].nxt=head[from];head[from]=tail;}
struct Tree{
int rev,val,fa,child[2],maxn,sumn;
}tree[MAXN];
void pushup(int x){
tree[x].maxn=max(tree[tree[x].child[0]].maxn,tree[tree[x].child[1]].maxn);
tree[x].maxn=max(tree[x].maxn,tree[x].val);
tree[x].sumn=tree[tree[x].child[0]].sumn+tree[tree[x].child[1]].sumn+tree[x].val;
}
void pushdown(int x){
if(tree[x].rev){
tree[x].rev^=1;tree[tree[x].child[0]].rev^=1;tree[tree[x].child[1]].rev^=1;
swap(tree[x].child[0],tree[x].child[1]);
}
}
bool isroot(int x){
return tree[tree[x].fa].child[0]!=x&&tree[tree[x].fa].child[1]!=x;
}
void Pushdown(int x){
if(!isroot(x))Pushdown(tree[x].fa);
pushdown(x);
}
void rotate(int x){
int y=tree[x].fa,z=tree[y].fa,l,r;
if(tree[y].child[0]==x) l=0;else l=1;r=l^1;
if(!isroot(y)){
if(tree[z].child[0]==y) tree[z].child[0]=x;
else tree[z].child[1]=x;
}
tree[tree[x].child[r]].fa=y;tree[y].fa=x;tree[x].fa=z;
tree[y].child[l]=tree[x].child[r];tree[x].child[r]=y;
pushup(y);pushup(x);
}
void splay(int x){
Pushdown(x);
while(!isroot(x)){
int y=tree[x].fa,z=tree[y].fa;
if(!isroot(y)){
if((tree[y].child[0]==x)^(tree[z].child[0]==y)) rotate(x);
else rotate(y);
}
rotate(x);
}
}
void access(int x){for(register int i=0;x;i=x,x=tree[x].fa){splay(x);tree[x].child[1]=i;pushup(x);}}
void makeroot(int x){access(x);splay(x);tree[x].rev^=1;}
void dfs(int u,int fa){
for(register int i=head[u];i;i=line[i].nxt){
int v=line[i].to;
if(v!=fa){
tree[v].fa=u;
dfs(v,u);
}
}
}
void init(){
scanf("%d",&n);
tree[0].maxn=-2147483647;
for(register int i=1;i<=n-1;i++){
scanf("%d%d",&from,&to);
add_line(from,to);add_line(to,from);
}
dfs(1,0);
for(register int i=1;i<=n;i++) scanf("%d",&tree[i].val),tree[i].maxn=tree[i].sumn=tree[i].val;
}
void solve(){
int Q,x,y;scanf("%d",&Q);
char opt[10];
while(Q--){
scanf("%s",&opt);
if(opt[1]=='H'){
scanf("%d%d",&x,&y);
splay(x);tree[x].val=y;pushup(x);
}else if(opt[1]=='M'){
scanf("%d%d",&x,&y);
makeroot(x);access(y);splay(y);printf("%d\n",tree[y].maxn);
}else{
scanf("%d%d",&x,&y);
makeroot(x);access(y);splay(y);printf("%d\n",tree[y].sumn);
}
}
}
int main(){
init();
solve();
return 0;
}