剑指 Offer II 053. 二叉搜索树中的中序后继

给定一棵二叉搜索树和其中的一个节点 p ，找到该节点在树中的中序后继。如果节点没有中序后继，请返回 null 。

节点 p 的后继是值比 p.val 大的节点中键值最小的节点，即按中序遍历的顺序节点 p 的下一个节点。

示例 1：

输入：root = [2,1,3], p = 1
输出：2
解释：这里 1 的中序后继是 2。请注意 p 和返回值都应是 TreeNode 类型。
示例 2：

输入：root = [5,3,6,2,4,null,null,1], p = 6
输出：null
解释：因为给出的节点没有中序后继，所以答案就返回 null 了。

提示：

树中节点的数目在范围 [1, 104] 内。
-105 <= Node.val <= 105
树中各节点的值均保证唯一。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/P5rCT8
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

充分利用二叉搜索树的性质，如果找一个节点的中序后继节点，只需要找到比该节点大一个的节点，定义 node 为每次递归的节点，result 为存储的结果，如果某时刻 node 的值大于 p 的值，说明 node 可能是 后继，存储，并让 node 往左走，看看有无更优的结果，如果 node 的值小于等于 p，说明找不到中序的后继，此时让 node 往右走就完了

/**
 * 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) {
        TreeNode node = root;
        TreeNode result = null;
        while (node != null) {
            if (node.val <= p.val) {
                node = node.right;
            } else {
                result = node;
                node = node.left;
            }
        }
        return result;
    }
    
}

如果不是二叉搜索树，就用 中序遍历进行求解，t 设置值即可

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    int t = 0;
    TreeNode next;
    public TreeNode inorderSuccessor(TreeNode root, TreeNode p) {
        next = null;
        inorder(root, p);
        return next;
    }
    public void inorder(TreeNode node, TreeNode p) {
        if (node == null) {
            return ;
        }
        
        inorder(node.left, p);
        if (t == 1) {
            next = node;
            t = 0;
        }
        if (node == p) {
            t = 1;
        }
        inorder(node.right, p);
    }
    
}