215. 采用计数排序思想解决数组中的第K个最大元素 排序

package leetcode;

import java.util.Arrays;

public class p215 {
    public static void main(String[] args) {

    }


    /**
     * 思路1：先使用快速排序进行排序，再返回第k大的数
     * 因为快速排序时间复杂度为O(n),所以不符合题目要求，但还是通过了
     */
    public static int findKthLargest2(int[] nums, int k) {
        Arrays.sort(nums);
        return nums[nums.length - k];
    }

    /**
     * 思路2：采用计数排序的思想，将小于0和大于0的数分开考虑。
     * 先遍历一次数组，分别求出最小值和最大值，再遍历一次数组，对小于0和大于0的数分别计数
     * 对于大于0的数，使用k从后往前减，对于小于0的数，从前往后减，当k<=0时即找到第k大的数
     *
     * 时间复杂度：O(n)
     * 空间复杂度：O(k) k为最大的数
     */
    public static int findKthLargest3(int[] nums, int k) {
        int max_p = -1; // 正数最大值
        int max_n = 0; // 负数最大值
        for (int i : nums) {
            if (i >= 0 && i > max_p) {
                max_p = i;
            } else if (i<0 && i < max_n) {
                max_n = i;
            }
        }
        int[] A = new int[max_p + 1]; // 计数正数
        int[] B = new int[(-1*max_n) + 1]; // 计数负数
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] >= 0) { // 分别对正数和负数进行计数
                A[nums[i]]++;
            }else {
                B[(-1)*nums[i]]++;
            }
        }
        for (int i=A.length-1;i>=0;i--){
            k-=A[i]; // 先计算正数
            if (k<=0){
                return i;
            }
        }
        for (int i=0;i<B.length;i++){
            k-=B[i]; // 再计算负数
            if (k<=0){
                return (-1*i); // 返回负数最大值
            }
        }
        return -1;
    }

}