leetcode刷题250天(74)——面试题 04.06. 后继者(二叉搜索树)

最新推荐文章于 2024-10-06 13:05:10 发布

ウルトラマン.

最新推荐文章于 2024-10-06 13:05:10 发布

阅读量121

点赞数

分类专栏： leetcode 文章标签： leetcode 算法面试

本文链接：https://blog.csdn.net/qq_35668477/article/details/124792918

版权

leetcode 专栏收录该内容

103 篇文章 1 订阅

订阅专栏

设计一个算法，找出二叉搜索树中指定节点的“下一个”节点（也即中序后继）。
如果指定节点没有对应的“下一个”节点，则返回null。

class Solution(object):
    def inorderSuccessor(self, root, p):
        result = root

        while root:
            # 当前结点值大于目标结点值
            # 说明当前结点值应该在目标结点值遍历序列的后面
            # 那就应该去左子树再去找找有没有更合适的
            # 有 那就换 没有 说明当前存的结点就是比目标结点大一号那个
            if root.val > p.val:
                result = root
                root = root.left
            # 目标值大于当前结点值
            # 往右子树再找找
            else:
                root = root.right

        return result if result.val > p.val else None