1. 程式人生 > >Leetcode 215.陣列中的第k個最大元素

Leetcode 215.陣列中的第k個最大元素

陣列中的第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

說明:

你可以假設 k 總是有效的,且 1 ≤ k ≤ 陣列的長度。

 

 1 class Solution {
 2     public int findKthLargest(int[] nums, int k) {
3 int begin=0; 4 int end=nums.length-1; 5 k=nums.length+1-k; 6 while(begin<end){ 7 int pos=partition(nums,begin,end); 8 if(pos==k-1) break; 9 else if(pos<k-1) begin=pos+1; 10 else end=pos-1; 11 } 12 return
nums[k-1]; 13 } 14 public int partition(int[] nums, int low, int high) {//分割槽 15 int key; 16 if (low > high) { 17 return 0; 18 } 19 key = nums[low]; // 用子表的第一個記錄做基準 20 while (low < high) { // 從表的兩端交替向中間掃描 21 while (low < high && nums[high] >= key)
22 high--; 23 if (low < high) 24 swap(nums,low,high);// 用比基準小的記錄替換低位記錄 25 while (low < high && nums[low] < key) 26 low++; 27 if (low < high) // 用比基準大的記錄替換高位記錄 28 swap(nums,low,high); 29 } 30 return low; 31 } 32 33 private void swap(int[] a, int i, int j) { 34 int t = a[i]; 35 a[i] = a[j]; 36 a[j] = t; 37 } 38 }