LeetCode230 二叉树中第K小的元素

最新推荐文章于 2023-10-29 20:29:47 发布

咸鱼在厦大

最新推荐文章于 2023-10-29 20:29:47 发布

阅读量136

点赞数

分类专栏： LeetCode Python 文章标签： leetcode 二叉树算法

本文链接：https://blog.csdn.net/Candylx/article/details/113184056

版权

LeetCode 同时被 2 个专栏收录

43 篇文章 0 订阅

订阅专栏

Python

38 篇文章 0 订阅

订阅专栏

LeetCode230 二叉树中第K小的元素

代码：

#生成器写法

class Solution:
    def mid_order(self, root):
        if not root: return
        yield from self.mid_order(root.left)
        yield root.val
        yield from self.mid_order(root.right)

    def kthSmallest(self, root: TreeNode, k: int) -> int:
        gen = self.mid_order(root)
        for _ in range(k - 1):
            next(gen)
        return next(gen)