215. 數組中的第K個最大元素。在未排序的數組中找到第 k 個最大的元素。請注意,你需要找的是數組排序後的第 k 個最大的元素,而不是第 k 個不同的元素。

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 ≤ 數組的長度。

  • 方法一:Arrays,自帶排序方法

class Solution {

    public int findKthLargest(int[] nums, int k) {

        Arrays.sort(nums);

        

        for(int i = nums.length-1,j=1; i>=0; i--,j++){

            if(j == k)

                return nums[i];

        }

        return -1;

    }

}

  • 方法二:快速排序

class Solution {

    Random random = new Random();

 

    public int findKthLargest(int[] nums, int k) {

        return quickSelect(nums, 0, nums.length - 1, nums.length - k);

    }

 

    public int quickSelect(int[] a, int l, int r, int index) {

        int q = randomPartition(a, l, r);

        if (q == index) {

            return a[q];

        } else {

            return q < index ? quickSelect(a, q + 1, r, index) : quickSelect(a, l, q - 1, index);

        }

    }

 

    public int randomPartition(int[] a, int l, int r) {

        int i = random.nextInt(r - l + 1) + l;

        swap(a, i, r);

        return partition(a, l, r);

    }

 

    public int partition(int[] a, int l, int r) { 

        int x = a[r], i = l - 1;

        for (int j = l; j < r; ++j) {

            if (a[j] <= x) {

                swap(a, ++i, j);

            }

        }

        swap(a, i + 1, r);

        return i + 1;

        }

 

    public void swap(int[] a, int i, int j) {

        int temp = a[i];

        a[i] = a[j];

        a[j] = temp;

    }

}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章