剑指 Offer II 053. 二叉搜索树中的中序后继
2022-03-13 本文已影响0人
Abeants
给定一棵二叉搜索树和其中的一个节点 p ,找到该节点在树中的中序后继。如果节点没有中序后继,请返回 null 。
节点 p 的后继是值比 p.val 大的节点中键值最小的节点,即按中序遍历的顺序节点 p 的下一个节点。
示例 1:
输入:root = [2,1,3], p = 1
输出:2
解释:这里 1 的中序后继是 2。请注意 p 和返回值都应是 TreeNode 类型。
示例 2:
输入:root = [5,3,6,2,4,null,null,1], p = 6
输出:null
解释:因为给出的节点没有中序后继,所以答案就返回 null 了。
提示:
树中节点的数目在范围 [1, 104] 内。
-105 <= Node.val <= 105
树中各节点的值均保证唯一。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/P5rCT8
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路及方法
- 下一个节点的值一定不会小于节点p的值,而是大于或等于节点p的值中的所有节点中值最小的一个
- 从根节点开始,每到达一个节点就比较根节点的值和节点p的值
- 如果当前节点的值小于或等于节点p的值,那么节点p的下一个节点应该在它的右子树
- 如果当前节点的值大于或等于节点p的值,那么当前节点有可能是p的下一个节点,此时当前节点的值比节点p的值大,但节点p的下一个节点是所有比它大的节点中值最小的一个,因此接下来前往当前节点的左子树,确定是否能找到值更小,但仍然大于节点p的值的节点
作者:angela-x
链接:https://leetcode-cn.com/problems/P5rCT8/solution/shi-jian-fu-za-du-wei-ceng-gao-de-jie-fa-fheg/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode inorderSuccessor(TreeNode root, TreeNode p) {
TreeNode res = null;
TreeNode cur = root;
while (cur != null) {
if (cur.val > p.val) {
res = cur;
cur = cur.left;
} else {
cur = cur.right;
}
}
return res;
}
}
结果如下: