LeetCode 215. 陣列中的第K個最大元素 Python3
阿新 • • 發佈:2018-12-09
在未排序的陣列中找到第 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])