洛谷P3379 【模板】最近公共祖先（LCA）

#include<bits/stdc++.h>
using namespace std;
const int maxn=500005;
inline void read(int &tmp)
{
    int x=1;char c=getchar();
    for(tmp=0;!isdigit(c);c=getchar()) if(c=='-') x=-1;
    for(;isdigit(c);tmp=tmp*10+c-48,c=getchar());
    tmp*=x;
}
struct edge{int to;edge *nex;}e[maxn<<1],*head[maxn];
int tot;
#define New(p) p=&e[++tot];
int n,T,root;
int deep[maxn],f[maxn][32];
inline void add(int x,int y) {edge *p;New(p);p->nex=head[x];p->to=y;head[x]=p;}
void dfs(int k)
{
    for(edge *i=head[k];i!=NULL;i=i->nex)
    {
        if(!deep[i->to])
        {
            deep[i->to]=deep[k]+1;
            f[i->to][0]=k;
            dfs(i->to);
        }
    }
}
void init()
{
    for(int j=1;(1<<j)<=n;++j)
        for(int i=1;i<=n;++i)
            if(f[i][j-1]) f[i][j]=f[f[i][j-1]][j-1];
}
int LCA(int a,int b)
{
    if(deep[a]<deep[b]) swap(a,b);
    int i;for(i=0;(1<<i)<=n;++i);--i;
    for(int j=i;j>=0;--j)
        if(deep[a]-(1<<j)>=deep[b]) a=f[a][j];
    if(a==b) return a;
    for(int j=i;j>=0;--j)
        if(f[a][j]&&f[a][j]!=f[b][j]) a=f[a][j],b=f[b][j];
    return f[a][0];
}
int main()
{
    read(n),read(T),read(root);
    for(int i=1,x,y;i<n;++i) read(x),read(y),add(x,y),add(y,x);
    deep[root]=1;f[root][0]=0;
    dfs(root);
    init();
    while(T--) {int a,b;read(a),read(b);printf("%d
",LCA(a,b));}
    return 0;
}