在二叉树中找到一个节点的后继节点

在二叉树中找到一个节点的后继节点

题目描述：

二叉树中一个节点的后继节点指的是，二叉树的中序遍历的序列中的下一个节点。

输入描述：

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

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

最后一行输入要询问的节点 node。

输出描述：

输出一个整数表示答案。(如果 node 是最后一个节点，则输出 0 )

示例1

输入

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

输出

0

备注：

$1\le n\le 5\ast 10^5$

$1\le fa,lch,rch,root,node\le n$

---

题解：

模拟，分情况讨论，假设待查找节点为 node ：

• 若 node 有右儿子，则后继节点为 node 右子树的最左节点；
• 若 node 无右儿子，则顺着 node 往上查找，当遇到一个父节点满足 bst[bst[node].fa].lch = node 时，则后继节点就是 bst[node].fa ，这样意味着建树时需要保存父节点。

代码：

#include <cstdio>

using namespace std;

const int N = 500010;

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

int n, rt;
int fa, lch, rch;
int node;

int left_most(int root) {
    if (!root) return root;
    while (bst[root].lch) root = bst[root].lch;
    return root;
}

int solve() {
    if (bst[node].rch) return left_most(bst[node].rch);
    while (bst[node].fa && bst[bst[node].fa].lch != node)
        node = bst[node].fa;
    return bst[node].fa;
}

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;
        bst[lch].fa = bst[rch].fa = fa;
    }
    scanf("%d", &node);
    printf("%d\n", solve());
    return 0;
}