LeetCode 215. 数组中的第K个最大元素 Python3

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Victordas/article/details/82623381

在未排序的数组中找到第 k 个最大的元素。请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。

示例 1: 输入: [3,2,1,5,6,4] 和 k = 2 输出: 5
示例 2: 输入: [3,2,3,1,2,4,5,5,6] 和 k = 4 输出: 4

说明:
你可以假设 k 总是有效的,且 1 ≤ k ≤ 数组的长度。

用Python写也太简单了吧。。刚开始还想复杂了

class Solution:
    def findKthLargest(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: int
        """
        nums.sort()

        """如果要求的是不重复的第四个最大位可以加上下面那句"""
        # nums = list(set(nums))

        if k in range(0, len(nums)+1):
            return(nums[-k])

猜你喜欢

转载自blog.csdn.net/Victordas/article/details/82623381