面试题 04.06. 后继者

题目表述

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

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

示例 1:

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

2
/
1 3

输出: 2

解题思路

只有两种情况

1. p没有右子树
2. p有右子树

如果有右子树的话，那么只需要直接根据中序遍历的规则，找到p的右子树的最左孩子即可

否则就递归向下找即可，用pre记录后继节点。

/**
 * 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(p.right != null){
            p = p.right;
            while(p.left != null){
                p = p.left;
            }
            return p;
        }

        TreeNode pre = null, cur = root;
        while(cur != p){
            if(p.val < cur.val){
                pre = cur;
                cur = cur.left;
            }else{
                cur = cur.right;
            }
        }
        return pre;
    }
}