树链剖分

我们先来看这样一个问题:
给出一棵树,每个节点有一个权值,每次操作可以修改某一节点的权值,或是询问任意两点间路径上的最大值或是权值和,要求复杂度为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;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值