leetcode:二叉搜索树节点的后继节点

题目来源:力扣

题目介绍：

设计一个算法，找出二叉搜索树中指定节点的“下一个”节点（也即中序后继）。
如果指定节点没有对应的“下一个”节点，则返回null。

审题：

根据二叉搜索树的特点，我们可以整理出判断后继节点的递归思路：

• 如果指定节点ｐ的值大于等于当前根节点的值，则节点ｐ的后继节点位于根节点右子树中（如果存在的话）．
• 如果制定节点ｐ的值小于当前根节点值，则节点ｐ的后继节点可能存在两种情况：节点ｐ是当前根节点左子树的最大节点，此时其后继节点为当前根节点，反之，其后继节点可以在左子树中递归查找得到．当节点ｐ的后继节点为当前根节点时，其在左子树的递归查找中返回null，因此可以容易判断当前节点后继节点是否是根节点．

java算法实现：

/**
 * 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) {
        if(root == null)
            return null;
        if(p.val >= root.val)
            return inorderSuccessor(root.right, p);
        else{
            TreeNode left = inorderSuccessor(root.left, p);
            return left == null ? root : left;
        }
    }
}