【Leetcode】285. Inorder Successor in BST

题目地址：

https://leetcode.com/problems/inorder-successor-in-bst/

给定一棵BST，再给定其中一节点$p$，求其中序遍历的后继。

首先后继肯定是比$p$大的，并且是所有比$p$大的节点里最小的那个。我们考虑应该如何找到这个后继。首先指针指向root，如果root小于等于$p$，那么root不可能是答案，左子树更不可能含答案，所以root向右走一步；否则如果root大于$p$，则用一个指针指向root父亲节点，然后跟在root后面，root向左走一步。之所以要用另一个指针跟在后面，是因为即使知道了root大于$p$，只能确定右子树不可能有答案，但root自己有可能是答案，所以要用一个指针跟在后面。当root走到null的时候程序结束，返回那个跟在后面的指针即可。代码如下：

public class Solution {
    public TreeNode inorderSuccessor(TreeNode root, TreeNode p) {
        TreeNode res = null;
        while (root != null) {
            if (root.val <= p.val) {
                root = root.right;
            } else {
                res = root;
                root = root.left;
            }
        }
        
        return res;
    }
}

class TreeNode {
    int val;
    TreeNode left, right;
    TreeNode(int x) {
        val = x;
    }
}

时间复杂度$O(h)$，空间$O(1)$。

算法正确性证明：
首先证明root一定在某个时刻正好走到了$p$的位置上。首先容易知道root总共也只能走有限步。并且，如果root小于等于$p$，要么root就在$p$上，要么$p$在root右子树上；否则$p$就在root的左子树上。所以root一定某个时刻正好在$p$上。
接着，如果$p$有右子树，那么root会走到右子树上，并且一路向左，显然最左边的节点就是答案。如果$p$无右子树，那么顺着$p$先向左上方走（如果$p$本身左上方没有节点就不用走了），再向右上方走一步，那么这里就是res停留的地方，其比$p$要大并且是$p$的后继（如果最后向右上方走一步到了null，则说明$p$本身就是最大值，其后继为null，算法也成立。关于这一点，可以考虑一下树根到$p$的路径就能推断出来）。