【Python】【难度：简单】Leetcode 面试题54. 二叉搜索树的第k大节点

最新推荐文章于 2023-05-01 20:14:35 发布

给阿姨倒杯卡布基诺

最新推荐文章于 2023-05-01 20:14:35 发布

阅读量176

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/thomashhs12/article/details/106299612

版权

leetcode 专栏收录该内容

260 篇文章 3 订阅

订阅专栏

给定一棵二叉搜索树，请找出其中第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 ≤ 二叉搜索树元素个数

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-di-kda-jie-dian-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def kthLargest(self, root, k):
        """
        :type root: TreeNode
        :type k: int
        :rtype: int
        """
        res=[]
        queue=[root]

        while queue:
            nxt=[]

            for node in queue:
                res.append(node.val)
                if node.left:
                    nxt.append(node.left)
                if node.right:
                    nxt.append(node.right)
            queue=nxt

        return sorted(res,reverse=True)[k-1]

执行结果：

通过

显示详情

执行用时 :44 ms, 在所有 Python 提交中击败了77.48%的用户

内存消耗 :20.6 MB, 在所有 Python 提交中击败了100.00%的用户