Leetcode组队学习任务

最新推荐文章于 2022-09-16 19:00:06 发布

好快的剑

最新推荐文章于 2022-09-16 19:00:06 发布

阅读量125

点赞数

原文链接：https://leetcode-cn.com/problems/kth-smallest-element-in-a-bst/solution/zhong-xu-bian-li-

版权

task12

给定一个二叉搜索树，编写一个函数 kthSmallest 来查找其中第 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)