在二叉树中找到两个节点的最近公共祖先(进阶)

在二叉树中找到两个节点的最近公共祖先(进阶)

题目描述

给定一棵二叉树，多次给出这棵树上的两个节点 o1 和 o2，请对于每次询问，找到 o1 和 o2 的最近公共祖先节点。

输入描述：

第一行输入两个整数 n 和 root，n 表示二叉树的总节点个数，root 表示二叉树的根节点。

以下 n 行每行三个整数 fa，lch，rch，表示 fa 的左儿子为 lch，右儿子为 rch。(如果 lch 为 0 则表示 fa 没有左儿子，rch同理)

第 n+2 行输入一个整数 m，表示询问的次数。

以下 m 行每行两个节点 o1 和 o2。

输出描述：

对于每组询问每行输出一个整数表示答案。

示例1

输入

8 1
1 2 3
2 4 5
4 0 0
5 0 0
3 6 7
6 0 0
7 8 0
8 0 0
4
4 5
5 2
6 8
5 8

输出

2
2
3
1

备注：

$1\le n\le 1000$

$1\le m\le 10^6$

$1\le fa,lch,rch,root,o_1,o_2\le n$

$o_1\ne o_2$

---

题解：

此题可以采用在线和离线两种写法，我选择的是在线解法：

在线（只查询当前一对节点）：

建树过程中保存节点的父节点，每次查询时，分别保存每个节点到根节点的序列，最后从尾到首同时遍历两个序列，最后一个相等的节点就是它们的最近公共祖先节点；

离线（提前处理所有节点对的最近公共祖先，查询时直接返回结果就行）：

处理比较麻烦，可参考书上该部分内容。。。

代码：

#include <cstdio>

using namespace std;

const int N = 1010;

struct BST {
    int fa, lch, rch;
} bst[N];

int n, rt;
int fa, lch, rch;
int m, o1, o2;
int preo1[N], preo2[N];
int n_o1, n_o2;

void get_par(int &root, int arr[], int& k) {
    while (root) {
        arr[k++] = root;
        root = bst[root].fa;
    }
}

int LCA(int &o1, int &o2) {
    n_o1 = n_o2 = 0;
    get_par(o1, preo1, n_o1);
    get_par(o2, preo2, n_o2);
    int i = n_o1 - 1, j = n_o2 - 1;
    while (i >= 0 && j >= 0) {
        if (preo1[i] == preo2[j]) --i, --j;
        else return preo1[i + 1];
    }
    if (i < 0) return preo1[i + 1];
    else return preo2[j + 1];
}

int main(void) {
    scanf("%d%d", &n, &rt);
    bst[rt].fa = 0;
    while (n--) {
        scanf("%d%d%d", &fa, &lch, &rch);
        bst[fa].lch = lch;
        bst[fa].rch = rch;
        if (lch) bst[lch].fa = fa;
        if (rch) bst[rch].fa = fa;
    }
    scanf("%d", &m);
    while (m--) {
        scanf("%d%d", &o1, &o2);
        printf("%d\n", LCA(o1, o2));
    }
    return 0;
}