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

leetcode 215 陣列中的第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:先排序再查詢

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;
}