题目描述:
给定一棵二叉搜索树和其中的一个节点 node ,找到该节点在树中的中序后继。
如果节点没有中序后继,请返回 null 。
一个结点 node 的中序后继是键值比 node.val大所有的结点中键值最小的那个。
你可以直接访问结点,但无法直接访问树。每个节点都会有其父节点的引用。节点定义如下:
class Node {
public int val;
public Node left;
public Node right;
public Node parent;
}
进阶:
你能否在不访问任何结点的值的情况下解决问题?
示例 1:
输入: tree = [2,1,3], node = 1
输出: 2
解析: 1 的中序后继结点是 2 。注意节点和返回值都是 Node 类型的。
示例 2:
输入: tree = [5,3,6,2,4,null,null,1], node = 6
输出: null
解析: 该结点没有中序后继,因此返回 null 。
示例 3:
输入: tree = [15,6,18,3,7,17,20,2,4,null,13,null,null,null,null,null,null,null,null,9], node = 15
输出: 17
示例 4:
输入: tree = [15,6,18,3,7,17,20,2,4,null,13,null,null,null,null,null,null,null,null,9], node = 13
输出: 15
提示:
-10^5 <= Node.val <= 10^5
1 <= Number of Nodes <= 10^4
树中各结点的值均保证唯一。
方法1:
主要思路:
(1)递归;
(2)若当前结点的右子树存在,则在右子树中去找最左边的结点;
(3)否则,向上遍历,既父节点存在,且父节点的右子树是当前子树,知道不满足这两个条件;
(4)返回当前子树的父节点即可;
/*
// Definition for a Node.
class Node {
public:
int val;
Node* left;
Node* right;
Node* parent;
};
*/
class Solution {
public:
Node* inorderSuccessor(Node* node) {
//右子树存在,则找右子树的最左结点
if(node->right){
node=node->right;
while(node->left){
node=node->left;
}
return node;
}
//否则,向上找,找到可能遍历过的根节点,该根节点作为其父节点的
while(node->parent&&node==node->parent->right){
node=node->parent;
}
return node->parent;
}
};