1. 程式人生 > >LeetCode 215. 陣列中的第K個最大元素 Python3

LeetCode 215. 陣列中的第K個最大元素 Python3

在未排序的陣列中找到第 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])