LCA板子题

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

#include<bits/stdc++.h>
using namespace std;
const int maxn=5e5+10;
int f[maxn][20],d[maxn],dist[maxn];
int head[maxn],ver[2*maxn],nxt[2*maxn],edge[2*maxn];
int tot,t;
int n,qq,r;
queue<int> q;
void add(int u,int v,int w)
{
    ver[++tot]=v;
    edge[tot]=w;
    nxt[tot]=head[u];
    head[u]=tot;
}
void bfs()
{
    q.push(r);
    d[r]=1;
    while(q.size())
    {
        int x=q.front();
        q.pop();
        for(int i=head[x]; i; i=nxt[i])
        {
            int y=ver[i];
            if(d[y]) continue;
            d[y]=d[x]+1;
            dist[y]=dist[x]+edge[i];
            f[y][0]=x;
            for(int j=1; j<=t; j++)
                f[y][j]=f[f[y][j-1]][j-1];
            q.push(y);
        }
    }
}
int lca(int x,int y)
{
    if(d[x]>d[y]) swap(x,y);
    for(int i=t; i>=0; i--)
        if(d[f[y][i]]>=d[x]) y=f[y][i];
    if(x==y) return x;
    for(int i=t; i>=0; i--)
        if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    return f[x][0];
}
int main()
{

    scanf("%d%d%d",&n,&qq,&r);
    t=(int)(log(n)/log(2))+1;
    for(int i=1; i<=n-1; i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        add(u,v,1);
        add(v,u,1);
    }
    bfs();
    for(int i=1; i<=qq; i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        printf("%d
",lca(a,b));
    }

}
原文地址:https://www.cnblogs.com/dongdong25800/p/11232348.html