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;

}
}