快速求树中某个节点的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