[算法练习及思路-程序员面试金典(Java解法)]No29.后继者

题号:no29

题目名:后继者

原题URL:https://leetcode-cn.com/problems/successor-lcci/

题目描述

设计一个算法，找出二叉搜索树中指定节点的“下一个”节点（也即中序后继）。

如果指定节点没有对应的“下一个”节点，则返回null。

示例

示例 1：

输入: root = [2,1,3], p = 1

  2
 / \
1   3

输出: 2

示例 2：

输入: root = [5,3,6,2,4,null,null,1], p = 6

      5
     / \
    3   6
   / \
  2   4
 /   
1

输出: null

思路

1.本来用的是中序遍历，放入到集合中，然后拿后面一个节点，这样速度太慢了

2.更新思路如下，目标节点和当前遍历节点比，有以下两种情况：1，目标节点大于等于当前节点，在右子树；2，小于当前节点，在左子树或者是他自己。

3.先对情况一进行递归遍历，然后遍历情况2，如果找不到，那就是他自己

解题代码

    public TreeNode inorderSuccessor(TreeNode root, TreeNode p) {
        //找得就是中序遍历的后继节点。那么二叉搜索树的中序遍历就是完全升序的
        //目标值大于当前值，说明目标值的后继肯定在右子树
        if (p.val >= root.val) {
            return inorderSuccessor(root.right, p);
        }
        // 否则就在其左子树，如果找不到，那就说明为空
        TreeNode node = inorderSuccessor(root.left, p);
        return node == null ? root : null;
    }