leetcode 215 陣列中的第k個最大的元素
阿新 • • 發佈:2018-12-04
在未排序的陣列中找到第 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 ≤ 陣列的長度。
解法1:先排序再查詢
public int findKthLargest(int[] nums, int k) { Arrays.sort(nums); return nums[nums.length - k]; }
解法2:使用快排的思想
核心思想是每次都要先找一箇中樞點Pivot,然後遍歷其他所有的數字,像這道題從小往大排的話,就把小於中樞點的數字放到左半邊,把大於中樞點的放在右半邊,這樣中樞點是整個陣列中第幾大的數字就確定了,雖然左右兩部分不一定是完全有序的,但是並不影響本題要求的結果,所以我們求出中樞點的位置,如果正好是k-1,那麼直接返回該位置上的數字;如果大於k-1,說明要求的數字在左半部分,更新右邊界,再求新的中樞點位置;反之則更新右半部分,求中樞點的位置。
public int findKthLargest(int[] nums, int k) { int left = 0; int right = nums.length - 1; while(true){ int pos = partation(nums, left, right); if(pos == k - 1) return nums[pos]; if(pos > k - 1){ right = pos - 1; }else{ left = pos + 1; } } } public int partation(int[] nums, int left, int right){ int privote = nums[left], l = left + 1, r = right; while(l <= r){ if(nums[l] < privote && nums[r] > privote){ swap(nums, l++, r--); } if(nums[l] >= privote) l++; if(nums[r] <= privote) r--; } swap(nums, left, r); return r; } private void swap(int[] nums, int i, int j){ int temp = nums[left]; nums[left] = nums[right]; nums[right] = temp; }