【题解】剑指 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 了。 

解题思路

本题的关键就是，并不能转化为数组来判断，因为平衡二叉树，如果拿TreeNode动态数组存储，则存储会破坏树的结构，如果只拿值存储，又无法返回节点，

 只能通过遍历实现，当当前节点大于给定节点的时候，记录当前节点，并且进入左子树，

                                当当前节点小于给定节点的时候，直接进入右子树

因为中序遍历中的下一个节点，对于左子树来说是父节点，对于右子树来说，是囊括右子树的最小左子树的父节点，代表的都是同一个父节点（这里有点绕）

 

题解

class Solution {
public:
    TreeNode* res = NULL;
    void dfs(TreeNode* root, TreeNode* p){
        if(!root)   return ;
        if(root->val>p->val){
            res = root;
            dfs(root->left,p);
        }else{
            dfs(root->right,p);
        }
    }
    TreeNode* inorderSuccessor(TreeNode* root, TreeNode* p) {
        dfs(root,p);
        return res;
    }
};