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

最新推荐文章于 2023-08-08 08:02:38 发布

菜菜2022

最新推荐文章于 2023-08-08 08:02:38 发布

阅读量47

点赞数

分类专栏：刷题 # leetcode

本文链接：https://blog.csdn.net/kz_java/article/details/118291396

版权

leetcode 同时被 2 个专栏收录

714 篇文章 5 订阅

订阅专栏

刷题

643 篇文章 5 订阅

订阅专栏

【题目】
给定一棵二叉搜索树，请找出其中第k大的节点。

示例 1:

输入: root = [3,1,4,null,2], k = 1
3
/
1 4

2
输出: 4

示例 2:

输入: root = [5,3,6,2,4,null,null,1], k = 3
5
/
3 6
/
2 4
/
1
输出: 4

限制：

1 ≤ k ≤ 二叉搜索树元素个数
【代码】
在这里插入图片描述

class Solution:
    def visit(self,root):
        if not root:
            return
        self.visit(root.right)
        self.ans.append(root.val)
        self.visit(root.left)
    def kthLargest(self, root: TreeNode, k: int) -> int:
        self.ans=[]
        self.visit(root)
        return self.ans[k-1]

【迭代法】
在这里插入图片描述

class Solution:
    def kthLargest(self, root: TreeNode, k: int) -> int:
        stack=[]
        node=root
        ans=[]
        while stack or node:
            while node:
                stack.append(node)
                node=node.right
            node=stack.pop()
            ans.append(node.val)
            if len(ans)==k:
                return ans[k-1]
            node=node.left
        return 0

【方法3：优化递归】
在这里插入图片描述

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