【剑指 Offer】剑指 Offer 54. 二叉搜索树的第k大节点

原创于 2022-03-11 17:25:03 发布 · 212 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #职场和发展

leetcode 同时被 2 个专栏收录

714 篇文章

订阅专栏

刷题

643 篇文章

订阅专栏

该博客介绍了两种不同的Python代码实现，用于遍历二叉树并找到第k个最大的节点值。第一种方法在寻找过程中先访问右子节点，然后更新k值，再访问左子节点；第二种方法同样先访问右子节点，但在找到第k个最大值后立即返回。这两种方法都在指定的时间和空间复杂度内完成了任务。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目
代码
执行用时：60 ms, 在所有 Python3 提交中击败了28.99% 的用户
内存消耗：18.9 MB, 在所有 Python3 提交中击败了52.56% 的用户
通过测试用例：91 / 91

class Solution:
    def visit(self,root):
        if not root:
            return
        self.visit(root.right)
        if self.k==0:
            return
        self.k-=1
        if self.k<=0:
            self.res=root.val
        self.visit(root.left)
    def kthLargest(self, root: TreeNode, k: int) -> int:
        self.k=k      
        self.res=None
        self.visit(root)
        return self.res

【方法2】
执行用时：52 ms, 在所有 Python3 提交中击败了69.89% 的用户
内存消耗：19 MB, 在所有 Python3 提交中击败了39.16% 的用户
通过测试用例：91 / 91

class Solution:
    def visit(self,root):
        if not root:
            return 
        self.visit(root.right)
        self.k-=1
        if self.k==0:
            self.ans=root.val
            return
        self.visit(root.left)

    def kthLargest(self, root: TreeNode, k: int) -> int:
        self.ans=0
        self.k=k
        self.visit(root)
        return self.ans