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

题目

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

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

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

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

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

输入输出样例
输入 #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
说明/提示
时空限制：1000ms,128M

数据规模：

对于30%的数据：N<=10，M<=10

对于70%的数据：N<=10000，M<=10000

对于100%的数据：N<=500000，M<=500000

样例说明：

该树结构如下：

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

第二次询问：3、2的最近公共祖先，故为4。

第三次询问：3、5的最近公共祖先，故为1。

第四次询问：1、2的最近公共祖先，故为4。

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

故输出依次为4、4、1、4、4。

lca 倍增法模版题；

详情见代码
AC代码：

#pragma GCC optimize(2)  //此题卡时间，要用o2优化
#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
const int N = 1e6 + 10;

vector<int> ve[N];
int depth[N];
int f[N][30];
ll pow2[30];

void POW2()
{
    pow2[0] = 1;
    for(int i = 1; i <= 29; i ++ ) pow2[i] = 2 * pow2[i - 1];
}

void dfs(int u, int fa)
{
    depth[u] = depth[fa] + 1;
    f[u][0] = fa;
    
    for(int i = 1; i <= 29; i ++ ) f[u][i] = f[f[u][i - 1]][i - 1];
    
    for(int i = 0; i <(int) ve[u].size(); i ++ )
    {
        int t = ve[u][i];
        
        if(t == fa) continue;
        
        dfs(t, u);
    }
}

int lca(int a, int b)
{
    if(depth[a] < depth[b]) swap(a, b);
    
    int dif = depth[a] - depth[b];
    
    for(int i = 29; i >= 0; i -- )
    {
    	if(pow2[i] <= dif)
    	{
    		a = f[a][i];
    		dif -= pow2[i];
		}
	}
    
    if(a == b) return a;
    
    for(int i = 29; i >= 0; i -- )
    {
        if(f[a][i] != f[b][i] && pow2[i] <= depth[a])
        {
            a = f[a][i], b = f[b][i];
        }
    }
    
    return f[a][0];
}

int main()
{
//	freopen("input.txt", "r",stdin);
	
	POW2();
    int n, m, s;
    int u, v;
    
    cin >> n >> m >> s;
    
    for(int i = 0; i < n - 1; i ++ )
    {
        scanf("%d %d", &u, &v);
        
        ve[u].push_back(v);
        ve[v].push_back(u);
    }
    
    dfs(s, 0);
    
    for(int i = 0; i < m; i ++ )
    {
        scanf("%d %d", &u, &v);
        printf("%d\n", lca(u, v));
    }
    
    return 0;
}