一棵树上有n个节点,编号分别为1到n,每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成 一些操作: I. CHANGE u t: 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 III. QSUM u v:询问从点u到点v的路径上的节点的权值和 注意:从点u到点v的路径上的节点包括u和v本身
bzoj中难得的模板题啊!日常做水题真是哈皮。树剖裸题。
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
struct node
{
int x,y,next;
}a[610000];int len,last[310000];
void ins(int x,int y)
{
len++;
a[len].x=x;a[len].y=y;
a[len].next=last[x];last[x]=len;
}
struct trnode
{
int l,r,lc,rc,mx,sum;
}tr[610000];int trlen;
void bt(int l,int r)
{
trlen++;int now=trlen;
tr[now].l=l;tr[now].r=r;
tr[now].lc=tr[now].rc=-1;
tr[now].mx=-99999;tr[now].sum=0;
if(l<r)
{
int mid=(l+r)/2;
tr[now].lc=trlen+1;bt(l,mid);
tr[now].rc=trlen+1;bt(mid+1,r);
}
}
int fa[31000],dep[31000],tot[31000],son[31000];
void pre_tree_node(int x)
{
tot[x]=1;son[x]=0;
for(int k=last[x];k;k=a[k].next)
{
int y=a[k].y;
if(y!=fa[x])
{
fa[y]=x;
dep[y]=dep[x]+1;
pre_tree_node(y);
if(tot[son[x]]<tot[y])son[x]=y;
tot[x]+=tot[y];
}
}
}
int id,ys[31000],top[31000];
void pre_tree_edge(int x,int tp)
{
ys[x]=++id;top[x]=tp;
if(son[x]!=0)pre_tree_edge(son[x],tp);
for(int k=last[x];k;k=a[k].next)
{
int y=a[k].y;
if(y!=fa[x] && y!=son[x])pre_tree_edge(y,y);
}
}
void change(int now,int x,int k)
{
if(tr[now].l==tr[now].r){tr[now].mx=tr[now].sum=k;return ;}
int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)/2;
if(x<=mid)change(lc,x,k);
else change(rc,x,k);
tr[now].mx=max(tr[lc].mx,tr[rc].mx);
tr[now].sum=tr[lc].sum+tr[rc].sum;
}
int findmax(int now,int l,int r)
{
if(tr[now].l==l && tr[now].r==r)return tr[now].mx;
int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)/2;
if(r<=mid)return findmax(lc,l,r);
else if(mid+1<=l)return findmax(rc,l,r);
else return max(findmax(lc,l,mid),findmax(rc,mid+1,r));
}
int getsum(int now,int l,int r)
{
if(tr[now].l==l && tr[now].r==r)return tr[now].sum;
int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)/2;
if(r<=mid)return getsum(lc,l,r);
else if(mid+1<=l)return getsum(rc,l,r);
else return getsum(lc,l,mid)+getsum(rc,mid+1,r);
}
int solve1(int x,int y)
{
int tx=top[x],ty=top[y],ans=-99999;
while(tx!=ty)
{
if(dep[tx]>dep[ty])swap(x,y),swap(tx,ty);
ans=max(ans,findmax(1,ys[ty],ys[y]));
y=fa[ty];ty=top[y];
}
if(dep[x]>dep[y])swap(x,y);
ans=max(ans,findmax(1,ys[x],ys[y]));
return ans;
}
int solve2(int x,int y)
{
int tx=top[x],ty=top[y],ans=0;
while(tx!=ty)
{
if(dep[tx]>dep[ty])swap(x,y),swap(tx,ty);
ans+=getsum(1,ys[ty],ys[y]);
y=fa[ty];ty=top[y];
}
if(dep[x]>dep[y])swap(x,y);
ans+=getsum(1,ys[x],ys[y]);
return ans;
}
char s[10];
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
ins(x,y);ins(y,x);
}
pre_tree_node(1);pre_tree_edge(1,1);bt(1,n);
for(int i=1;i<=n;i++)
{
int x;
scanf("%d",&x);
change(1,ys[i],x);
}
int t;
scanf("%d",&t);
while(t--)
{
int x,y;
scanf("%s%d%d",s+1,&x,&y);
if(s[2]=='H')change(1,ys[x],y);
if(s[2]=='M')printf("%d\n",solve1(x,y));
if(s[2]=='S')printf("%d\n",solve2(x,y));
}
return 0;
}