Java实现 LeetCode 215. 数组中的第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 ≤ 数组的长度。

class Solution {
    public int findKthLargest(int[] nums, int k) {
        if(nums.length < 2){
            return nums.length == 1 ? nums[0] : -1;
        }
        return countingSort(nums, k);
    }

    private int countingSort(int[] nums, int k){
        int max = 0;
        int min = 0;
        for(int i = 0; i < nums.length; i++){
            if(nums[i] > max){
                max = nums[i];
            }

            if(nums[i] < min){
                min = nums[i];
            }
        }

        int length = (max - min) + 1;
        int[] newArray = new int[length];
        //这个数组记录的是与最小的值相差的某位的数量
        for(int i = 0; i < nums.length; i++){
            newArray[nums[i] - min]++;
        }

        int j = 0;
        for(int i = newArray.length - 1; i >= 0; i--){
            //这样最大值就出来了，先从最大值开始
            if(newArray[i] > 0){
                j = newArray[i] + j;
                if(j >= k){
                    return i + min;
                }
            }
        }
        return -1;
    }
}