Leetcode 215.Kth Largest Element in an Array

通常遇到第 K 个大的元素这样的问题, 考虑排序类的问题.

本题的题目就是问在这个数组中第 K 大的元素

方法一

最简单直接的做法当然是将该数组排序.

这种方法的时间复杂度为 O(NlogN), 空间复杂度为 O(1)

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

方法二

利用小根堆来完成, Java 中的 PriorityQueue 来实现, 可以先将数组的前 k 个元素构造成一个小根堆, 然后依次将数组后的每个元素和小根堆的队首元素进行比较, 如果大于队首元素就入队, 否则就跳过该元素. 最后这个小根堆中的元素就是该数组中的最大的 K 个元素.因为 PriorityQueue 的默认顺序是从小到大, 所以其队首元素是第 K 大的元素.

这种方法的时间复杂度为 O(NlogK), 空间复杂度为 O(1)

class Solution {
    public int findKthLargest(int[] nums, int k) {
        PriorityQueue<Integer> pq = new PriorityQueue<>();
        int index = 0;
        for (; index < k; index++) {
            pq.add(nums[index]);
        }
        while (index < nums.length) {
            if (nums[index] > pq.peek()) {
                pq.poll();
                pq.add(nums[index++]);
            } else {
                index++;
            }
        }
        return pq.peek();
    }
}

方法三

利用快排的思想, 选择一个数来作为基准, 一趟快排下来, 如果该数的位置为 nums.length - k, 那么该数就是答案, 如果该数的位置小于 nums.length - k, 那么在这个数的左边进行快排, 否则在这个数的右边进行快排. 

这种方法的时间复杂度为 O(N), 空间复杂度为 O(1)

class Solution {
    public int findKthLargest(int[] nums, int k) {
        int left = 0, right = nums.length - 1;
        k = nums.length - k;
        while(left <= right) {
            int res = partition(nums, left, right);
            if (res == k) {
                break;
            } else if (res < k) {
                left = res + 1;
            } else {
                right = res - 1;
            }
        }
        return nums[k];
    }
    
    private int partition(int[] arr, int left, int right) {
        int pivot = left - 1;
        int index = left;
        while(index <= right) {
            if (arr[index] <= arr[right]) {
                swap(arr, ++pivot, index++);
            } else {
                index++;
            }
        }
        return pivot;
    }
    
    private void swap(int[] arr, int index1, int index2) {
        int temp = arr[index1];
        arr[index1] = arr[index2];
        arr[index2] = temp;
    }
}

对于基本排序的实现一定要熟悉, 需要进一步学习的是 Arrays.sort() 的实现原理与 PriorityQueue 的相关实现.