1. 題目

在未排序的數組中找到第 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 ≤ 數組的長度。

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/kth-largest-element-in-an-array
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 快排解題

• 參考：尋找數組內第K大的元素
• 類似題目：LeetCode 973. 最接近原點的 K 個點（排序/優先隊列/快排）

class Solution { //C++ public:int findKthLargest(vector<int>& nums, int k) {k = nums.size()-k;//排序后的位置return findKthL(nums,k,0,nums.size()-1);} private:void selectMid(vector<int>& nums, int left, int right){int mid = left+((right-left)>>1);if(nums[mid] > nums[right])swap(nums[mid],nums[right]);if(nums[left] > nums[right])swap(nums[left], nums[right]);if(nums[mid] > nums[left])swap(nums[mid], nums[left]);}int findKthL(vector<int>& nums, int &k, int left, int right){selectMid(nums,left,right);//三數取中int p = nums[left];int i = left, j = right;while(i < j){while(i < j && nums[j] > p)j--;swap(nums[i], nums[j]);while(i < j && nums[i] <= p)i++;swap(nums[i], nums[j]);}if(i == k)return nums[i];else if(i < k)return findKthL(nums,k,i+1,right);elsereturn findKthL(nums,k,left,i-1);} };

class Solution { //C++ ,簡化版 public:int findKthLargest(vector<int>& nums, int k) {return findk(nums,0,nums.size()-1, nums.size()-k);}int findk(vector<int>& nums, int l, int r, int k){int p = nums[l];int i = l, j = r;while(i < j){while(i < j && nums[j] > p)j--;while(i < j && nums[i] <= p)i++;swap(nums[i],nums[j]);}swap(nums[i], nums[l]);if(i == k)return nums[i];if(i < k)return findk(nums, i+1, r, k);elsereturn findk(nums, l, i-1, k);} };

56 ms 9.9 MB

python3 解答

class Solution:# py3def findKthLargest(self, nums: List[int], k: int) -> int:def findk(l, r, k):p = nums[l];i = lj = rwhile i < j:while i < j and nums[j] > p:j -= 1while i < j and nums[i] <= p:i += 1t = nums[i]nums[i] = nums[j]nums[j] = tt = nums[i]nums[i] = nums[l]nums[l] = tif i == k:return nums[i]elif i < k:return findk(i+1, r, k)else:return findk(l, i-1, k)return findk(0,len(nums)-1, len(nums)-k)

1032 ms 19.7 MB

總結

以上是生活随笔為你收集整理的LeetCode 215. 数组中的第K个最大元素（快速排序）的全部內容，希望文章能夠幫你解決所遇到的問題。

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