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

题目描述
如题，给定一棵有根多叉树，请求出指定两个点直接最近的公共祖先。

输入格式
第一行包含三个正整数 N,M,SN,M,S，分别表示树的结点个数、询问的个数和树根结点的序号。

接下来 N-1N−1 行每行包含两个正整数 x, yx,y，表示 xx 结点和 yy 结点之间有一条直接连接的边（数据保证可以构成树）。

接下来 MM 行每行包含两个正整数 a, ba,b，表示询问 aa 结点和 bb 结点的最近公共祖先。

输出格式
输出包含 MM 行，每行包含一个正整数，依次为每一个询问的结果。

输入输出样例
输入 #1复制
5 5 4
3 1
2 4
5 1
1 4
2 4
3 2
3 5
1 2
4 5
输出 #1复制
4
4
1
4
4
说明/提示
对于 30%30% 的数据，N\leq 10N≤10，M\leq 10M≤10。

对于 70%70% 的数据，N\leq 10000N≤10000，M\leq 10000M≤10000。

对于 100%100% 的数据，N\leq 500000N≤500000，M\leq 500000M≤500000。

样例说明：

该树结构如下：

第一次询问：2, 42,4 的最近公共祖先，故为 44。

第二次询问：3, 23,2 的最近公共祖先，故为 44。

第三次询问：3, 53,5 的最近公共祖先，故为 11。

第四次询问：1, 21,2 的最近公共祖先，故为 44。

第五次询问：4, 54,5 的最近公共祖先，故为 44。

故输出依次为 4, 4, 1, 4, 44,4,1,4,4。

#include <bits/stdc++.h>

using namespace std;
const int N=5e5+5;
int n,m,s,x,y,dep[N],fa[N][25];
vector<int> a[N];

int read()//快读
{
    char c;
    int f=1,x=0;
    c=getchar();
    while(!isdigit(c))
    {
        if(c=='-')
            f=-1;
        c=getchar();
    }
    while(isdigit(c))
    {
        x=x*10+c-'0';
        c=getchar();
    }
    return x*f;
}
void dfs(int x,int father)
{
    dep[x]=dep[father]+1;
    fa[x][0]=father;
    for(int i=1;(1<<i)<=dep[x];i++)//x节点的深度允许跳几个
    {
        fa[x][i]=fa[fa[x][i-1]][i-1];
    }
    for(int i=0;i<a[x].size();i++)
    {
        if(a[x][i]!=father)
            dfs(a[x][i],x);//相当于假设建树
    }
}
int lca(int u,int v)
{
    if(dep[u]<dep[v])
        swap(u,v);//默认x深度大
    //int depth=dep[u]-dep[v];//深度差
    for(int i=20;i>=0;i--)//第一步，调节至同深度
    {
        //从大到小，正整数拆分原理
        if(dep[u]-(1<<i)>=dep[v])
            u=fa[u][i];
    }
    if(u==v)//如果相同，则return
        return u;
    for(int i=20;i>=0;i--)//第二步，求最近公共祖先的下一层节点
    {
        //从大到小，正整数拆分原理
        if(fa[u][i]!=fa[v][i])
        {
            u=fa[u][i];
            v=fa[v][i];
        }
    }
    return fa[u][0];//返回LCA
}
int main()
{
    n=read();
    m=read();
    s=read();//根节点
    for(int i=0;i<n-1;i++)
    {
        x=read(),y=read();
        a[x].push_back(y);
        a[y].push_back(x);
    }
    dfs(s,0);//预处理 把0看做根
    while(m--)
    {
        x=read(),y=read();
        printf("%d\n",lca(x,y));
    }
    return 0;
}