LintCode 5. 第k大元素 Python

原创于 2022-08-04 20:05:01 发布 · 1.2k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #职场和发展

LintCode 刷题笔记专栏收录该内容

13 篇文章

订阅专栏

本文介绍了如何使用Python简单地解决LintCode上的第k大元素问题，通过列表排序获取第k大的元素，时间复杂度为O(n log n)，但实际操作中效率高。

LintCode题库第五题：第k大元素

我们可以用比较简单的方法，对列表进行从大到小的排序，然后获取元素

该方案时间复杂度为O(1)

用Python实现，代码如下：

class Solution:
    """
    @param k: An integer
    @param nums: An array
    @return: the Kth largest element
    """
    def kth_largest_element(self, k, nums):
        # write your code here
        return list(sorted(nums,reverse=True))[k-1]