剑指 Offer II 053. 二叉搜索树中的中序后继

最新推荐文章于 2024-10-08 10:36:00 发布

Sasakihaise_

最新推荐文章于 2024-10-08 10:36:00 发布

阅读量2.1k

点赞数

分类专栏：剑指offer 文章标签： leetcode BST

本文链接：https://blog.csdn.net/Sasakihaise_/article/details/124801587

版权

剑指offer 专栏收录该内容

42 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何在二叉搜索树中找到一个给定节点的中序后继节点。初始算法是遍历右子树，而改进后的算法考虑了当给定节点的右子树非空时的特殊情况，直接从右子树的最小节点开始查找。通过这两种方法，可以高效地在BST中找到目标节点的中序后继。

摘要由CSDN通过智能技术生成

剑指 Offer II 053. 二叉搜索树中的中序后继

【分析】根据BST的性质，如果root.val <= p.val，说明后继肯定在右子树，去右子树找就行了，否则，root是后继，但是不一定是最小的后继，所以还要去left找。

/**
 * 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 ans = null;
        while(root != null){
            if(root.val <= p.val) root = root.right;
            else{
                ans = root;
                root = root.left;
            }
        }
        return ans;
    }
}

【改进】如果p的右子树不为空，那么直接去p右边找就行了。

/**
 * 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 ans = null;
        if(p.right != null){
            p = p.right;
            while(p != null){
                ans = p;
                p = p.left;
            }
        }else{
            while(root != null){
                if(root.val <= p.val) root = root.right;
                else{
                    ans = root;
                    root = root.left;
                }
            }
        }
        return ans;
    }
}