[模板] 洛谷 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

说明/提示

对于 30% 的数据，N≤10，M≤10。

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

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

代码及注释

#include <cmath>
#include <cstdio>
#include <algorithm>
#include <iostream>
using namespace std;
#define MAXN 500050

//快速读入
inline int read() {
    int f = 0, x = 0; char ch = getchar();
    while (!isdigit(ch)) f = (ch == 45), ch = getchar();
    while ( isdigit(ch)) x = (x << 1) + (x << 3) + (ch ^ 48), ch = getchar();
    return f ? (~x + 1) : x;
}

struct edge {
    int next, to;
}e[MAXN << 1];
int head[MAXN << 1], num = 0;

inline void addedge(int from, int to) {
    e[++num].next = head[from];
    e[num].to = to;
    head[from] = num;
}

int n, m, S, lg[MAXN] = {0};
int fa[MAXN][21], dep[MAXN];

//对每一点计算深度以及其倍增处理
void dfs(int s, int f) {
    fa[s][0] = f, dep[s] = dep[f] + 1;

    //对每一点进行倍增处理
    for (int i = 1; i <= lg[dep[s]]; i++)
        fa[s][i] = fa[fa[s][i - 1]][i - 1];
    
    for (int i = head[s]; i; i = e[i].next) 
        if (e[i].to != f)
            dfs(e[i].to, s);
}

int LCA(int x, int y) {
    if (dep[x] < dep[y]) swap(x, y);
    
    //让x跳到与y同一层
    while (dep[x] > dep[y])
        x = fa[x][lg[dep[x] - dep[y]]];
    
    if (x == y) return x;

    //x, y跳到从上往下第一个不相同的层，那么此层的上一层便是LCA
    for (int i = lg[dep[x]]; i >= 0; i--)
        if (fa[x][i] != fa[y][i])
            x = fa[x][i], y = fa[y][i];
    
    return fa[x][0];
}

int main() {
    n = read(), m = read(); S = read();
    for (int i = 1; i < n; i++) {
        int x = read(), y = read();
        addedge(x, y);
        addedge(y, x);
    }

    //计算log2(n)
    for (int i = 2; i <= n; i++) 
        lg[i] = lg[i - 1] + ((1 << (lg[i - 1] + 1)) == i);
    
    dfs(S, 0);

    for(int i = 1; i <= m; i++)
        cout << LCA(read(), read()) << endl;
    
    return 0;
}