Leetcode 285. Inorder Successor in BST

最新推荐文章于 2022-01-16 05:43:31 发布

原创最新推荐文章于 2022-01-16 05:43:31 发布 · 341 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#bst #recursion #dfs

Leetcode 专栏收录该内容

278 篇文章

订阅专栏

博客介绍了二叉搜索树（BST）中序遍历的两种方法。方法一是模拟中序遍历，新建全局栈，按特定顺序放入节点，遍历栈得到中序遍历顺序，时间和空间复杂度均为n；方法二则巧妙运用了BST的性质，时间复杂度为n，空间复杂度为1。

在这里插入图片描述
方法1: simulate an inorder traversal。新建一个全局stack。对于每个node，我们先放node.right，再放node，最后放node.left。最后我们遍历stack，这样pop出来的顺序正好就是inorder traversal的顺序。时间复杂n，空间复杂n。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    Stack<TreeNode> stack = new Stack<>();
    public TreeNode inorderSuccessor(TreeNode root, TreeNode p) {
        if(root == null) return null;
        inorder(root);
        while(!stack.isEmpty()){
            TreeNode curr = stack.pop();
            if(curr == p && !stack.isEmpty()) return stack.pop();
        }
        return null;
    }
    public void inorder(TreeNode root){
        if(root == null){
            return;
        } 
        inorder(root.right);
        stack.push(root);
        inorder(root.left);
    }
}

方法2: 上面我的方法并没有用到bst这个性质，方法二非常巧妙的运用了bst。时间复杂n，空间复杂1.

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    // Solution: BST
    // Use the BST characteristic to find the smallest node
    // larger than p
    // Time complexity: O(n) -- traverse all nodes
    // Space complexity: O(1)
    public TreeNode inorderSuccessor(TreeNode root, TreeNode p) {
        if (root == null || p == null) {
            return null;
        }
        TreeNode successor = null;
        while (root != null) {
            if (root.val > p.val) {
                successor = root;
                root = root.left;
            } else {
                root = root.right;
            }
        }
        return successor;        
    }
}