在二叉树中找到一个节点的后继节点
题目描述:
二叉树中一个节点的后继节点指的是,二叉树的中序遍历的序列中的下一个节点。
输入描述:
第一行输入两个整数 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 ≤ n ≤ 5 ∗ 1 0 5 1 \leq n \leq 5*10^5 1≤n≤5∗105
1 ≤ f a , l c h , r c h , r o o t , n o d e ≤ n 1 \leq fa,lch,rch,root,node \leq n 1≤fa,lch,rch,root,node≤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;
}