LeetCode 二叉搜索树中第k小的数

最新推荐文章于 2022-09-08 09:10:42 发布

jay_wonder

最新推荐文章于 2022-09-08 09:10:42 发布

阅读量240

点赞数

文章标签：算法 python leetcode

本文链接：https://blog.csdn.net/jay_wonder/article/details/104852601

版权

中序遍历 + 一个计数变量即可解决
在这里插入图片描述

class Solution:
    def kthSmallest(self, root: TreeNode, k: int) -> int:
        count = 0
        res = float('inf')
        def helper(root, k):
            nonlocal count
            nonlocal res
            if not root:
                return
            helper(root.left, k)
            count += 1
            if count == k:  #找到第k小的数
                res = root.val
                return
            helper(root.right, k)
        helper(root, k)
        return res