lintcode 448. 二叉查找树的中序后继

最新推荐文章于 2022-01-22 23:30:18 发布

MissZhou要努力

最新推荐文章于 2022-01-22 23:30:18 发布

阅读量236

点赞数

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.csdn.net/zhou_yujia/article/details/101114417

版权

leetcode 专栏收录该内容

89 篇文章 0 订阅

订阅专栏

本来以为既然都是找后继都是一样的

结果都！不！一！样！

查找的时候怎么用val比较呢？

考虑如果当前root的值比查找的小于或等于：淘汰掉左子树就好

要是需要找到左子树了，那么递归下去同样的方法，如果找不到，那么说明当前这个根节点是要找的

class Solution {
public:
    /*
     * @param root: The root of the BST.
     * @param p: You need find the successor node of p.
     * @return: Successor of p.
     */
    TreeNode * inorderSuccessor(TreeNode * root, TreeNode * p) {
        if (root == NULL) return NULL;
        if (root->val <= p->val) {
            return inorderSuccessor(root->right, p);
        } else {
            TreeNode* left = inorderSuccessor(root->left, p);
            return left?left:root;
        }
    }
};

个人喜欢这个非递归的，先根据p的val找到等于val的节点

然后，根据next的性质，那么是父节点，那么是右左左左左

父节点在上次找=val的时候找到的更新

class Solution {
public:
    /*
     * @param root: The root of the BST.
     * @param p: You need find the successor node of p.
     * @return: Successor of p.
     */
    TreeNode * inorderSuccessor(TreeNode * root, TreeNode * p) {
        if (root == NULL) return NULL;
        TreeNode* fa = NULL;
        while(root && root->val != p->val) {
            if(root->val < p->val) root = root->right;
            else if (root->val > p->val) {
                fa = root;
                root = root->left;
            }
        }
        TreeNode* current = root;
        if(root == NULL) return NULL;
        if(root->right == NULL) return fa;
        else {
            root = root->right;
            while(root->left) root= root->left;
            return root;
        }
    }
};