我们先来看这样一个问题:
给出一棵树,每个节点有一个权值,每次操作可以修改某一节点的权值,或是询问任意两点间路径上的最大值或是权值和,要求复杂度为O(nlogn)n为点数。
由于有修改操作,使得不能使用预处理(倍增数组)来实现快速查询,树链剖分就可以通过把树剖分成一些链,对于链上的节点,可以用数据结构来维护(如线段树),树链剖分就是使得剖分出来的链可以保证复杂度的剖分方法,这里介绍重链剖分:
对于每个节点,记它的子树节点个数为siz[u],siz[]最大的儿子为重儿子,其他儿子为轻儿子,当前节点与重儿子的连边为重边,当前节点与轻儿子的连边为轻边,一条链是重链当且仅当这条链是由重边组成的,有如下性质:
1.对于任意轻边,记父节点为u,子节点为v,有siz[v]
<
<script type="math/tex" id="MathJax-Element-190"><</script>siz[u]/2;
2.任意节点到根的路径上,最多有logn条轻边;
3.任意节点到根的路径上,最多有logn条重链(要使重链条数最多,一定是重链轻边交替走,而轻边最多有logn条,所以重链最多有logn条)
这样,对于重链上的点维护线段树,为了方便,可以只建一棵线段树,只要让重链上的点都在同一段区间上即可。
预处理需要两次dfs,
第一次,预处理siz[],fa[],H_son[],dep[]数组。
第二次,预处理top[]表示i所在的重链的顶端节点,id[],注意,由于要使在同一条重链上的节点在线段树上的某一段区间,所以每次都要先遍历重儿子。
单点修改
直接在线段树上修改就可以了。
路径查询
可以参照LCA倍增思想,当两个节点的top[]值不同时,每次让top[]深的往上跳,u=fa[top[u]],同时把u到top[u]这条重链上的信息与当前信息合并,直到两个节点在同一条重链上,最后不要忘记把最后一条重链上的信息与当前信息合并,以查询最大值为例,代码如下:
int get_max(int x,int y){
int Max=-1e9;
for(;top[x]!=top[y];x=fa[top[x]]){
if(dep[top[x]]<dep[top[y]])swap(x,y);
Max=max(Max,query_max(1,id[top[x]],id[x]));
}
if(dep[x]<dep[y])swap(x,y);
return max(Max,query_max(1,id[y],id[x]));
}
由于有前面的三条性质,所以可以证明单次操作复杂度为O(log^2(n));
给出代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 30006
using namespace std;
struct data{
int l,r,max,sum,key;
}tre[maxn*4];
int n,tot,tem,tet,lnk[maxn],nxt[maxn*2],son[maxn*2],dep[maxn],H_son[maxn],top[maxn],siz[maxn],fa[maxn],id[maxn];
bool vis[maxn];
int _read(){
int sum=0,p;char ch=getchar();
while((!(ch>='0'&&ch<='9'))&&(ch!='-'))ch=getchar();
if(ch=='-')p=-1,ch=getchar();else p=1;
while(ch>='0'&&ch<='9')sum=sum*10+ch-48,ch=getchar();
return sum*p;
}
void add(int x,int y){
nxt[++tot]=lnk[x];son[tot]=y;lnk[x]=tot;
}
void build(int p,int l,int r){
int mid=(l+r)>>1;
tre[p].l=l;tre[p].r=r;tre[p].max=-1e9;tre[p].key=tre[p].sum=0;
if(l==r)return;
build(p<<1,l,mid);build(p<<1|1,mid+1,r);
}
void update(int p,int pos,int k){
if(pos>tre[p].r||pos<tre[p].l)return;
if(tre[p].l==tre[p].r){
tre[p].key=tre[p].max=tre[p].sum=k;
return;
}
update(p<<1,pos,k);update(p<<1|1,pos,k);
tre[p].max=max(tre[p<<1].max,tre[p<<1|1].max);
tre[p].sum=tre[p<<1].sum+tre[p<<1|1].sum;
}
int query_max(int p,int l,int r){
if(l>tre[p].r||r<tre[p].l)return -1e9;
if(l<=tre[p].l&&r>=tre[p].r)return tre[p].max;
return max(query_max(p<<1,l,r),query_max(p<<1|1,l,r));
}
int query_sum(int p,int l,int r){
if(l>tre[p].r||r<tre[p].l)return 0;
if(l<=tre[p].l&&r>=tre[p].r)return tre[p].sum;
return query_sum(p<<1,l,r)+query_sum(p<<1|1,l,r);
}
void dfs1(int x){
vis[x]=0;siz[x]=1;
for(int j=lnk[x];j;j=nxt[j]) if(vis[son[j]]){
fa[son[j]]=x;dep[son[j]]=dep[x]+1;
dfs1(son[j]);
if(siz[son[j]]>siz[H_son[x]])H_son[x]=son[j];
siz[x]+=siz[son[j]];
}
}
void dfs2(int x,int lst){
vis[x]=0;top[x]=lst;id[x]=++tem;
if(H_son[x])dfs2(H_son[x],lst);
for(int j=lnk[x];j;j=nxt[j]) if(vis[son[j]]&&(son[j]!=H_son[x]))dfs2(son[j],son[j]);
}
int get_max(int x,int y){
int Max=-1e9;
for(;top[x]!=top[y];x=fa[top[x]]){
if(dep[top[x]]<dep[top[y]])swap(x,y);
Max=max(Max,query_max(1,id[top[x]],id[x]));
}
if(dep[x]<dep[y])swap(x,y);
return max(Max,query_max(1,id[y],id[x]));
}
int get_sum(int x,int y){
int sum=0;
for(;top[x]!=top[y];x=fa[top[x]]){
if(dep[top[x]]<dep[top[y]])swap(x,y);
sum+=query_sum(1,id[top[x]],id[x]);
}
if(id[x]>id[y])swap(x,y);
return sum+query_sum(1,id[x],id[y]);
}
int main(){
freopen("count.in","r",stdin);
freopen("count.out","w",stdout);
n=_read();
for(int i=1,x,y;i<n;i++)x=_read(),y=_read(),add(x,y),add(y,x);
memset(vis,1,sizeof(vis));
dfs1(1);
memset(vis,1,sizeof(vis));
dfs2(1,1);
build(1,1,tem);
for(int i=1,x;i<=n;i++)x=_read(),update(1,id[i],x);
tet=_read();
while(tet--){
char s[30];
scanf("%s",s);int x=_read(),y=_read();
if(s[0]=='C')update(1,id[x],y);else
if(s[1]=='M')printf("%d\n",get_max(x,y));else
if(s[1]=='S')printf("%d\n",get_sum(x,y));
}
return 0;
}