题目传送门:233333
直接把那棵树剖开来,剖成一条条的链,然后直接用两棵线段树维护就好了啊
不会树链剖分的出门左拐。
然后你会撞到墙上2333.
树剖思路:
设size为以该节点为根的子树的结点个数,对于每一个不是叶子节点的节点,在儿子中选出一个size最大的作为重儿子,那条边叫做重链。
之后再一次dfs,优先搜索重儿子,整理出搜索顺序之后就会发现,连在一起的重链的编号在搜索顺序中是连续的!!
于是就可以用线段树维护了
------------------------------------------
以上纯属口胡,详情见百度。
#include <cstdio>
#include <algorithm>
#include <iostream>
#include <cstring>
#define rep(j,k,l) for (int j=k;j<=l;j++)
#define N 30005
using namespace std;
struct _233{
int l,r,sum,m_x;
} tr[N*5];
int n,T,cnt,a[N],to[N*2],ne[N*2],st[N];
int deep[N],size[N],fa[N],son[N],top[N],id[N],dfn[N];
void add(int k,int l,int p){
to[p]=l;
ne[p]=st[k];
st[k]=p;
}
void dfs1(int rt,int dad){
size[rt]=1;
fa[rt]=dad;
deep[rt]=deep[dad]+1;
son[rt]=0;int _=0;
for (int i=st[rt];i!=0;i=ne[i])
if (to[i]!=dad){
dfs1(to[i],rt);
size[rt]+=size[to[i]];
if (size[to[i]]>_){
_=size[to[i]];
son[rt]=to[i];
}
}
}
void dfs2(int rt,bool qaz){
if (qaz) top[rt]=top[fa[rt]];
else top[rt]=rt;
dfn[++cnt]=rt;
id[rt]=cnt;
if (son[rt]!=0) dfs2(son[rt],1);
for (int i=st[rt];i!=0;i=ne[i])
if (to[i]!=fa[rt]&&to[i]!=son[rt])
dfs2(to[i],0);
}
void stree(int k,int l,int r){
if (l==r){
tr[k].sum=a[dfn[l]];
tr[k].m_x=a[dfn[l]];
return;
}
tr[k].l=++cnt;
tr[k].r=++cnt;
stree(tr[k].l,l,(l+r)/2);
stree(tr[k].r,(l+r)/2+1,r);
tr[k].sum=tr[tr[k].l].sum+tr[tr[k].r].sum;
tr[k].m_x=max(tr[tr[k].l].m_x,tr[tr[k].r].m_x);
return;
}
void change(int k,int l,int r,int o,int p){
if (o<l||o>r) return;
if (l==r){
tr[k].sum=p;
tr[k].m_x=p;
return;
}
change(tr[k].l,l,(l+r)/2,o,p);
change(tr[k].r,(l+r)/2+1,r,o,p);
tr[k].sum=tr[tr[k].l].sum+tr[tr[k].r].sum;
tr[k].m_x=max(tr[tr[k].l].m_x,tr[tr[k].r].m_x);
}
int _m_x(int k,int l,int r,int o,int p){
if (o>r||p<l) return -40000;
if (o<=l&&r<=p) return tr[k].m_x;
return max(_m_x(tr[k].l,l,(l+r)/2,o,p),_m_x(tr[k].r,(l+r)/2+1,r,o,p));
}
int askmax(int x,int y){
int ans=-40000;
for (;top[x]!=top[y];x=fa[top[x]]){
if (deep[top[x]]<deep[top[y]]) swap(x,y);
ans=max(ans,_m_x(1,1,n,id[top[x]],id[x]));
}
ans=max(ans,_m_x(1,1,n,min(id[x],id[y]),max(id[x],id[y])));
return ans;
}
int _sum_(int k,int l,int r,int o,int p){
if (o>r||p<l) return 0;
if (o<=l&&r<=p) return tr[k].sum;
return _sum_(tr[k].l,l,(l+r)/2,o,p)+_sum_(tr[k].r,(l+r)/2+1,r,o,p);
}
int asksum(int x,int y){
int ans=0;
for (;top[x]!=top[y];x=fa[top[x]]){
if (deep[top[x]]<deep[top[y]]) swap(x,y);
ans=ans+_sum_(1,1,n,id[top[x]],id[x]);
}
ans=ans+_sum_(1,1,n,min(id[x],id[y]),max(id[x],id[y]));
return ans;
}
int main(){
scanf("%d",&n);
rep(i,1,n-1){
int k,l;
scanf("%d%d",&k,&l);
add(k,l,2*i-1);add(l,k,2*i);
}
rep(i,1,n) scanf("%d",&a[i]);
dfs1(1,0);
dfs2(1,0);
cnt=1;
stree(1,1,n);
scanf("%d",&T);
while (T--){
char ch=getchar();
while (ch<'A'||ch>'Z') ch=getchar();
if (ch=='C'){
ch=getchar();
ch=getchar();
ch=getchar();
ch=getchar();
ch=getchar();
int k,l;
scanf("%d%d",&k,&l);
change(1,1,n,id[k],l);
}
else{
ch=getchar();
ch=getchar();
ch=getchar();
int k,l;
scanf("%d%d",&k,&l);
if (ch=='X') printf("%d\n",askmax(k,l));
else printf("%d\n",asksum(k,l));
}
}
}