LeetCode 510. 二叉搜索树中的中序后继 II

具体思路：

直接按照具体的思路去找，但是第一次看到这种题，直接给节点不给根节点的；

分情况讨论即可；

具体代码：

// // Definition for a Node.
// class Node {
// public:
//     int val;
//     Node* left;
//     Node* right;
//     Node* parent;
// };


class Solution {
public:
    Node* inorderSuccessor(Node* node) {
        Node* ret=nullptr;
        find(node,ret);
        return ret;
    }

    void find(Node* root,Node* &ret){
        //如果右节点存在，直接找右节点的最左子节点；
        if(root->right){
            root=root->right;
            while(root->left){
                root=root->left;
            }
            ret=root;
            return;
        }
        //向上回溯，如果是上一个节点的左子节点；
        if(root->parent!=nullptr&&root->parent->left==root){
            ret=root->parent;
            return;
        }
        //向上回溯，如果是上一个节点的右子节点；
        while(root->parent!=nullptr&&root->parent->right==root){
            root=root->parent;
        }
        if(root->parent!=nullptr)
            ret=root->parent;
        return;
    }
};