题目:
一棵树上有n个节点,编号分别为1到n,每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成
一些操作: I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 I
一些操作: 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”的操作,每行输出一个整数表示要求输出的结果。
思路:直接套lct就行了
代码:
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<list>
#include<numeric>
using namespace std;
#define LL long long
#define ULL unsigned long long
#define INF 0x3f3f3f3f
#define mm(a,b) memset(a,b,sizeof(a))
#define PP puts("*********************");
template<class T> T f_abs(T a){ return a > 0 ? a : -a; }
template<class T> T gcd(T a, T b){ return b ? gcd(b, a%b) : a; }
template<class T> T lcm(T a,T b){return a/gcd(a,b)*b;}
// 0x3f3f3f3f3f3f3f3f
// 0x3f3f3f3f
const int maxn=3e4+50;
const int maxe=2*maxn;
struct Edge{
int to,next;
}edge[maxe];
int head[maxn],tot;
int ch[maxn][2],key[maxn],pre[maxn],rev[maxn];
int Max[maxn],sum[maxn];
bool rt[maxn];//标记节点是不是splay的根
int n;//n个节点,从1开始计数
void init(){
tot=0;
for(int i=0;i<=n;i++){
head[i]=-1;
rt[i]=true;
ch[i][0]=ch[i][1]=0;
pre[i]=0;
rev[i]=0;
sum[i]=0;
}
Max[0]=-2000000000;//0节点一定要初始化好,因为要push_up
}
void addedge(int u,int v){
edge[tot].to=v;
edge[tot].next=head[u];
head[u]=tot++;
}
void dfs(int u,int fa){
pre[u]=fa;
Max[u]=key[u];
sum[u]=key[u];
for(int i=head[u];i!=-1;i=edge[i].next){
int v=edge[i].to;
if(v!=fa)
dfs(v,u);
}
}
void update_rev(int x){
if(!x) return;
swap(ch[x][0],ch[x][1]);
rev[x]^=1;
}
void push_down(int x){
if(rev[x]){
update_rev(ch[x][0]);
update_rev(ch[x][1]);
rev[x]=0;
}
}
void push_up(int x){
Max[x]=key[x];
Max[x]=max(Max[x],Max[ch[x][0]]);
Max[x]=max(Max[x],Max[ch[x][1]]);
sum[x]=key[x]+sum[ch[x][0]]+sum[ch[x][1]];
}
void rotate(int x){
int y=pre[x],d=ch[y][1]==x;
ch[y][d]=ch[x][!d];
pre[ch[y][d]]=y;
pre[x]=pre[y];
pre[y]=x;
ch[x][!d]=y;
if(rt[y]) rt[y]=false,rt[x]=true;
else ch[pre[x]][ch[pre[x]][1]==y]=x;
push_up(y);
}
//P函数先将根结点到x的路径上所有的结点的标记逐级下放
void P(int x){
if(!rt[x]) P(pre[x]);
push_down(x);
}
//将x旋转到根
void splay(int x){
P(x);
while(!rt[x]){
int f=pre[x],ff=pre[f];
if(rt[f]) rotate(x);
else if((ch[ff][1]==f)==(ch[f][1]==x))
rotate(f),rotate(x);
else
rotate(x),rotate(x);
}
push_up(x);
}
//将x到根的路径变成首选边
int Access(int x){
int y=0;
for(;x;x=pre[y=x]){
splay(x);
rt[ch[x][1]]=true;
rt[ch[x][1]=y]=false;
push_up(x);
}
return y;
}
//判断是否是同根(真实的树,非splay)
bool judge(int u,int v){
while(pre[u]) u=pre[u];
while(pre[v]) v=pre[v];
return u==v;
}
//让x成为它所在树的根
void make_root(int x){
Access(x);
splay(x);
update_rev(x);
}
//调用后u是原来u和v的lca,v和ch[u][1]分别存着lca的2个儿子
//(原来u和v所在的2颗子树)
void lca(int &u,int &v){
Access(v),v=0;
while(u){
splay(u);
if(!pre[u]) return;
rt[ch[u][1]]=true;
rt[ch[u][1]=v]=false;
push_up(u);
u=pre[v=u];
}
}
//如果u,v不在同一棵子树中,则连接这2棵子树
void link(int u,int v){
if(judge(u,v)){
printf("-1\n");
return;
}
make_root(u);
pre[u]=v;
}
//使u成为u所在树的根,并且将v和它父亲的边断开
void cut(int u,int v){
if(u==v||!judge(u,v)){
printf("-1\n");
return;
}
make_root(u);
Access(v);
splay(v);
pre[ch[v][0]]=0;
rt[ch[v][0]]=true;
ch[v][0]=0;
push_up(v);
}
int main(){
int Q,u,v;
char str[20];
while(~scanf("%d",&n)){
init();
for(int i=1;i<n;i++){
scanf("%d%d",&u,&v);
addedge(u,v);
addedge(v,u);
}
for(int i=1;i<=n;i++)
scanf("%d",&key[i]);
dfs(1,0);
scanf("%d",&Q);
while(Q--){
scanf("%s%d%d",str,&u,&v);
if(strcmp(str,"CHANGE")==0){
Access(u);
splay(u);
key[u]=v;
push_up(u);
}
else if(strcmp(str,"QMAX")==0){
make_root(u);
Access(v);
splay(v);
printf("%d\n",Max[v]);
}
else{
make_root(u);
Access(v);
splay(v);
printf("%d\n",sum[v]);
}
}
}
return 0;
}