题目描述
如题,给定一棵有根多叉树,请求出指定两个点直接最近的公共祖先。
输入格式
第一行包含三个正整数 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;
}