求LCA最近公共祖先的在线倍增算法模板_C++

时间:2024-05-07 07:50:46

  倍增求 LCA 是在线的,而且比 ST 好写多了,理解起来比 ST 和 Tarjan 都容易,于是就自行脑补吧,代码写得容易看懂

  关键理解 f[i][j] 表示 i 号节点的第 2j 个父亲,也就是往上走 2个节点

  求 LCA 的时候先倍增让两点深度一样,再倍增求

  另外丢两个链接,这两个有详细讲解

    ST 算法 http://www.cnblogs.com/hadilo/p/5837517.html

    Tarajan 算法 http://www.cnblogs.com/hadilo/p/5840390.html

  可能代码缩进不是很好看,因为我的 Emacs 用的默认缩进

 #include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std; const int N=,L=;
int m,first[N],next[N],d[N],f[N][L];
inline void dfs(int x,int dep)
{
d[x]=dep;
m=max(m,dep);
for (int i=first[x];i;i=next[i]) dfs(i,dep+);
}
int log2(int x)
{
int k=;
while (x>)
{
x>>=;
k++;
}
return k;
}
int main()
{
int i,j,n,s,x,y,root;
scanf("%d",&n);
for (i=;i<=n;i++)
{
scanf("%d",&f[i][]);
if (!f[i][]) root=i;
next[i]=first[f[i][]];
first[f[i][]]=i;
}
dfs(root,);
s=log2(m);
for (j=;j<=s;j++)
for (i=;i<=n;i++) f[i][j]=f[f[i][j-]][j-];
scanf("%d",&n);
while (n--)
{
scanf("%d%d",&x,&y);
if (d[x]<d[y]) swap(x,y);
s=log2(d[x]-d[y]);
while (d[x]>d[y])
{
if (d[x]-(<<s)>=d[y]) x=f[x][s];
s--;
}
s=log2(d[x]);
while (s>-)
{
if (f[x][s]!=f[y][s])
{
x=f[x][s];
y=f[y][s];
}
s--;
}
printf("%d\n",x==y?x:f[x][]);
}
return ;
}