數組中第k個最大元素

給定整數數組 nums 和整數 k，請返回數組中第 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提示：1 <= k <= nums.length <= 104 -104 <= nums[i] <= 104

題解

class Solution:def findKthLargest(self, nums: List[int], k: int) -> int:#parttion是按從小到大劃分的，如果讓index左邊為前n-k個小的數，則index右邊為前k個大的數self.topk_split(nums, len(nums)-k, 0, len(nums)-1) #把k換成len(nums)-kreturn nums[len(nums)-k]def partition(self, nums, left, right):pivot = nums[left]#初始化一個待比較數據i,j = left, rightwhile(i < j):while(i<j and nums[j]>=pivot): #從后往前查找，直到找到一個比pivot更小的數j-=1nums[i] = nums[j] #將更小的數放入左邊while(i<j and nums[i]<=pivot): #從前往后找，直到找到一個比pivot更大的數i+=1nums[j] = nums[i] #將更大的數放入右邊#循環結束，i與j相等nums[i] = pivot #待比較數據放入最終位置 return i #返回待比較數據最終位置def topk_split(self, nums, k, left, right):#尋找到第k個數停止遞歸，使得nums數組中index左邊是前k個小的數，index右邊是后面n-k個大的數if (left<right):index = self.partition(nums, left, right)if index==k:return elif index < k:self.topk_split(nums, k, index+1, right)else:self.topk_split(nums, k, left, index-1)

總結

以上是生活随笔為你收集整理的python 数组中第k个最大元素的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。