倍增lca模板

注意fa数组和depth数组一定要先初始化。

#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <string.h>
#include <vector>
using namespace std;
int fa[10005][50],depth[10005],lg[10005];
vector<int>g[10005];
void dfs(int now,int fath)
{
    depth[now]=depth[fath]+1;fa[now][0]=fath;
    for(int i=1;(1<<i)<=depth[now];i++)
        fa[now][i]=fa[fa[now][i-1]][i-1];
    int len=g[now].size();
    for(int i=0;i<len;i++)
    {
        int v=g[now][i];
        if(v!=fath)dfs(v,now);
    }
}
int lca(int x,int y)
{
    if(depth[x]<depth[y]) swap(x,y);
    while(depth[x]>depth[y]) x=fa[x][lg[depth[x]-depth[y]]-1];
    if(x==y) return x;
    for(int i=lg[depth[x]-1];i>=0;i--)
    {
        if(fa[x][i]!=fa[y][i])
        {
            x=fa[x][i];
            y=fa[y][i];
        }
    }
    return fa[x][0];
}
int pan[10005];
int main()
{
    int n,m;
    cin>>n;
    while(n--)
    {
        int u,v,a,b;
        scanf("%d",&m);
        memset(depth,0,sizeof(depth));
        memset(fa,0,sizeof(fa));
        memset(pan,0,sizeof(pan));
        for(int i=0;i<=m;i++) g[i].clear();
        for(int i=1;i<=m-1;i++)
        {
            scanf("%d %d",&u,&v);
            pan[v]++;
            g[u].push_back(v);
        }
        for(int i=1;i<=m;i++) if(pan[i]==0) {fa[i][0]=0;dfs(i,0);break;};
        for(int i=1;i<=m+1;i++) lg[i]=lg[i-1]+(1<<lg[i-1]==i);
        scanf("%d %d",&a,&b);
        printf("%d\n",lca(a,b));
    }
    return 0;
}