285. Inorder Successor in BST

这个题直接in-order是O(n)。。要求应该是lgn之类的。

分情况讨论:
有右节点,go right,然后无限往左到,直到左边是NULL,就停止,停止的地方就是答案。

没有右节点,就要traverse了。规则是,找到第一个大于P的点,找不到就是NULL,找到就保存下来。

当前点大于P,保存当前点,然后走左边;
小于P,走右边;
等于P停止,返还保存的点。

public class Solution 
{
    TreeNode prev = null;
    public TreeNode inorderSuccessor(TreeNode root, TreeNode p) 
    {
        if(root == null) return null;
        
        if(p.right != null)
        {
            TreeNode temp = p.right;
            while(temp.left != null) temp = temp.left;
            return temp;
        }
        else helper(root,p.val);
        return prev;
    }
    
    public void helper(TreeNode root, int val)
    {
        if(root.val == val) return;
        if(root.val < val) helper(root.right,val);
        else
        {
            prev = root;
            helper(root.left,val);
        }
    }
}

貌似可以不判断有没有右节点,上来就直接干也OK。二刷再说。。




二刷。

对于BST来说,in-order就是从小到大排列,找next child,就是大于P中最小的数。

从ROOT开始利用BST的大小特性往P逼近,如果当前NODE的值大于P,就记录下来,并且缩小当前值(go left);如果小于P,说明P可能在parent的右支,或者当前的右支的某一个左支:
前者返还记录的Parent,因为往右遍历的所有值都小于P,不会再找到一个大于P的值并记录;
后者会在右支的某一个点大于P,记录那个点,以那个点为ROOT,往左找,就又回到了一开始的大小判断。

。。说的很乱,但是思路很简单。

O(lgn),或者O(height)跟高度有关。

也可以直接遍历。。就O(n) time, O(n) space

public class Solution {
    TreeNode parent = null;
    public TreeNode inorderSuccessor(TreeNode root, TreeNode p) {
        if (root == null || p == null) return null;
        
        TreeNode res = null;
        TreeNode temp = root;
        
        while (temp != null) {
            if (temp.val > p.val) {
                res = temp;
                temp = temp.left;
            } else {
                temp = temp.right;
            }
        }
        
        return res;
        
    }
}
原文地址:https://www.cnblogs.com/reboot329/p/5962696.html