树上 LCA

树上LCA

倍增法

设 $f_{i,j}$ 为第 $i$ 个点往上跳 $2^j$ 个点所到的点,显然 $f_{i,0}$ 为第 $i$ 个点的父亲。

对于点 $u,v$,我们只需要将它们调整至同一深度,然后一起往上跳(不能重合)直到不能再跳,则最后它们的父亲就是 $lca(u,v)$。

我们可以 $dfs$ 预处理出 $f_{i,0}$ 和 $deep_i$。时间复杂度为 $O(n)$。

对于 $f_{i,j}$,可以看做 $i$ 号点向上跳 $2^{j-1}$ 个点后再跳 $2^{j-1}$ 个点得到的,因为 $2^{j-1} \times 2 = 2^j = 2^{j-1} + 2^{j-1}$。

所以得到递推式:

$f_{i,j}=f_{f_{i,j-1},j-1}$

预处理 $f$ 数组复杂度为 $O(n log n)$

查询复杂度为 $O(log n)$

代码

#include<bits/stdc++.h>
using namespace std;
int n,m,s;
vector<int>g[500005];
int f[500005][30],deep[500005];
void dfs(int x,int fa){
    f[x][0]=fa;
    deep[x]=deep[fa]+1;
    for(auto v:g[x]){
        if(v==fa)continue;
        dfs(v,x);
    }
}
int lca(int a,int b){
    if(deep[a]<deep[b])swap(a,b);
    for(int i=22;i>=0;i--){
        if(deep[f[a][i]]>=deep[b])a=f[a][i];
    }
    if(a==b)return a;
    for(int i=22;i>=0;i--){
        if(f[a][i]!=f[b][i]){
            a=f[a][i];
            b=f[b][i];
        }
    }
    return f[a][0];
}
int main(){
    ios::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
    cin>>n>>m>>s;
    for(int i=1;i<n;i++){
        int u,v;
        cin>>u>>v;
        g[u].push_back(v);
        g[v].push_back(u);
    }
    dfs(s,0);
    for(int j=1;(1<<j)<=n;j++){
        for(int i=1;i<=n;i++){
            f[i][j]=f[f[i][j-1]][j-1];
        }
    }
    while(m--){
        int a,b;
        cin>>a>>b;
        cout<<lca(a,b)<<"\n";
    }
    return 0;
}

P3379 【模板】最近公共祖先(LCA)