其他分享
首页 > 其他分享> > 快速求树中某个节点的k级祖先

快速求树中某个节点的k级祖先

作者:互联网

问题如题目所示。

方法有很多,本文说三种:

树上倍增/LCA魔改 O(n logn)

预处理得到fa数组,倍增向上跳跃,求祖先即可。

int fa[N][21],dep[N];
void dfs_lca(int u,int father){
    dep[u]=dep[father]+1;
    fa[u][0]=father;
    for(int i=1;(1<<i)<=dep[u];i++)
        fa[u][i]=fa[fa[u][i-1]][i-1];
    for(int i=h[u];i!=-1;i=pre[i]){
        int j=to[i];
        if(j!=father){
             dfs_lca(j,u);
        }
    }
}
int  get_fa(int u,int k){
//     if(k==0) return u;
    int temp=dep[u]-k;
    for(int i=20;i>=0;i--)
        if(dep[fa[u][i]]>=temp)
            u=fa[u][i];
    return u;
}

利用DFS栈 O(N)

在 dfs 时维护 dfs 栈,那么访问到点

标签:fa,祖先,top,求树中,father,dfs,int,节点
来源: https://www.cnblogs.com/kingwz/p/16573935.html