每日一道算法题LeetCode125：Kth Largest Element in an Array（数组中的第K个最大元素）

题目

在未排序的数组中找到第 k 个最大的元素。
请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。

分析

这个题的思路很简单，将数组排序，然后返回倒数 K 位置的元素即可，所以排序算法的选择就很关键了。

题解

快排

先使用快排，将数组排序，然后直接返回倒数 K 位置的元素：

	public static int findKthLargest(int[] nums, int k) {
        quickSort(nums,0,nums.length-1);
        return(nums[nums.length-k]);
    }
    public static void quickSort(int[] nums, int l, int r){
        if(l>=r) return;
        int i=l-1, j=r+1;
        int flag = nums[(l+r)/2];

        while(i<j){
            do i++ ; while(nums[i] < flag);
            do j-- ; while(nums[j] > flag);
            if(i<j){
                int temp = nums[i];
                nums[i] = nums[j];
                nums[j] = temp;
            }
        }

        quickSort(nums,l,j);
        quickSort(nums,j+1,r);
    }

其实这个代码等同下面（Array.sort底层使用的是双轴快排），hhh：

	public int findKthLargest(int[] nums, int k) {
        Arrays.sort(nums);
        return nums[nums.length - k];
    }

最小堆

维护一个最小堆，然后pop出 len-k 个元素，这时堆顶的元素就是我们要的。
相同的，维护一个最大堆，然后pop出 k-1 个元素，这时，堆顶元素是结果。

    public static int findKthLargest(int[] nums, int k) {
        int len = nums.length;
        PriorityQueue<Integer> maxHeap = new PriorityQueue<>(len);
        for (int num : nums) {
            maxHeap.add(num);
        }
        for (int i = 0; i < len-k; i++) {
            maxHeap.poll();
        }
        return maxHeap.peek();
    }

总结

这道题不是很难，正好把排序算法回顾一下，可以看看我之前写的java实现十大经典排序算法（一）和java实现十大经典排序算法（二）。