树链剖分专题

基本定义
deep:深度
size:一个节点的子树的节点数
son:重儿子(size最大的)
fa:节点的父亲
top:节点所在链的最高节点
基本思路
第一次dfs找到节点所有的基本信息;
第二次dfs沿着重儿子造成重链,以其他节点造新的重链;
于是树就被转化为链了。
然后再用其他数据结构优化
为了使所有的链连在一起我们可以给所有的点一个编号。
原理
由于一个点在被查询的过程中可以走重链或轻链,而如果都走轻链也只会走O(logn)次。

例题
CF 326 E . Duff in the Army

在一棵n个节点的树中,有m个人(编号为1到m),给出每个人在树上的节点位置。有q个询问,对于每个询问v,u,a ,输出k,并从小到大输出该路径上编号最小的a个人的编号(若路径上不足a个人,有多少个人就输出多少个,k为输出的数量)。
1 ≤ n, m, q ≤ 10^5,1 ≤ a ≤ 10。


int dep[M],fa[M],sz[M],son[M],top[M];
int Re[M],id[M];
void dfs(int x,int pre=0){
    sz[x]=1;dep[x]=dep[pre]+1;
    for(int i=0;i<G[x].size();i++){
        int y=G[x][i];
        if(y!=pre){
            fa[y]=x;
            dfs(y,x);sz[x]+=sz[y];
            if(sz[y]>sz[son[x]])son[x]=y;
        }
    }
}
void rdfs(int x,int f){
    top[x]=f;

    //给树上的点编号 
    Re[id[x]=++num]=x;

    if(son[x])rdfs(son[x],f);
    for(int i=0;i<G[x].size();i++){
        int y=G[x][i];
        if(y!=son[x]&&y!=fa[x])rdfs(y,y);
    }
}
//如果要找树上的从a到b的信息需要一条一条链的找
void Query(int a,int b){
    //不在一条链
    while(top[a]!=top[b]){
        //从最低的点开始跳
        if(dep[top[a]]<dep[top[b]])swap(a,b);
        //搜集整条链的信息
        query(id[top[a]],id[a]);
        //再跳出该链
        a=fa[top[a]];
    }if(id[a]>id[b])swap(a,b);
    query(id[a],id[b]);
}

综上

#include<bits/stdc++.h>
#define PB(a,b) G[a].push_back(b)
using namespace std;
const int M=1e5+5;
vector<int>S[M],G[M];
int a,b,c,n,m,q,num;
int dep[M],fa[M],sz[M],son[M],top[M];
int Re[M],id[M],T[12][M<<2],mx,ans[M];
void dfs(int x,int pre=0){
    sz[x]=1;dep[x]=dep[pre]+1;
    for(int i=0;i<G[x].size();i++){
        int y=G[x][i];
        if(y!=pre){
            fa[y]=x;
            dfs(y,x);sz[x]+=sz[y];
            if(sz[y]>sz[son[x]])son[x]=y;
        }
    }
}
void rdfs(int x,int f){
    top[x]=f;
    Re[id[x]=++num]=x;
    if(son[x])rdfs(son[x],f);
    for(int i=0;i<G[x].size();i++){
        int y=G[x][i];
        if(y!=son[x]&&y!=fa[x])rdfs(y,y);
    }
}
void build(int L=1,int R=num,int p=1){
    int &len=T[0][p];
    if(L==R){
        int nw=Re[L];
        for(int i=0;i<S[nw].size();i++){
            T[++len][p]=S[nw][i];
        }
        return;
    }int mid=R+L>>1,p1=p<<1,p2=p<<1|1, a=1,b=1;
    build(L,mid,p1);build(mid+1,R,p2);
    int l1=T[0][p1],l2=T[0][p2];
    while(len<10&&a<=l1&&b<=l2){
        if(T[a][p1]<T[b][p2])T[++len][p]=T[a++][p1];
        else T[++len][p]=T[b++][p2];
    }
    while(len<10&&a<=l1)T[++len][p]=T[a++][p1];
    while(len<10&&b<=l2)T[++len][p]=T[b++][p2];
}
void query(int l,int r,int L=1,int R=num,int p=1){
    if(r<L||R<l)return;
    if(l<=L&&R<=r){
        for(int i=1;i<=min(T[0][p],mx);i++)
            ans[++ans[0]]=T[i][p];
        return;
    }int mid=L+R>>1;
    query(l,r,L,mid,p<<1);
    query(l,r,mid+1,R,p<<1|1);
}

inline void solve(){
    int a,b,c;
    scanf("%d%d%d",&a,&b,&c);
    mx=c;ans[0]=0;
    while(top[a]!=top[b]){
        if(dep[top[a]]<dep[top[b]])swap(a,b);
        query(id[top[a]],id[a]);
        a=fa[top[a]];
    }if(id[a]>id[b])swap(a,b);
    query(id[a],id[b]);
    sort(ans+1,ans+1+ans[0]);
    ans[0]=min(ans[0],c);
    for(int i=0;i<=ans[0];i++)printf("%d ",ans[i]);
    puts("");
}
int main(){
    scanf("%d%d%d",&n,&m,&q);
    for(int i=1;i<n;i++){
        scanf("%d %d",&a,&b);
        PB(a,b);PB(b,a);
    }for(int i=1;i<=m;i++){
        scanf("%d",&a);
        if(S[a].size()<10)S[a].push_back(i);
    }
    dfs(1);
    rdfs(1,1);
    build();
    while(q--)solve();
    return 0;
}
  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值