Leetcode215. Kth Largest Element in an Array

Leetcode215. Kth Largest Element in an Array

Find the kth largest element in an unsorted array. Note that it is the kth largest element in the sorted order, not the kth distinct element.

Example 1:

Input: [3,2,1,5,6,4] and k = 2
Output: 5

Example 2:

Input: [3,2,3,1,2,4,5,5,6] and k = 4
Output: 4

Note:
You may assume k is always valid, 1 ≤ k ≤ array’s length.

解法一 暴力法

使用快排从大到小排序，将第 k 个数返回。

直接使用 java 提供的排序算法，又因为默认是从小到大排序，所以将倒数第 k 个数返回即可。

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

解法二 双指针

借鉴快速排序的分区算法

① 随机选择一个分区点，左边都是大于分区点的数，右边都是小于分区点的数。左部分的个数记做 m。
②如果 k == m + 1，我们把分区点返回即可。
③如果 k > m + 1，说明第 k大数在右边，我们在右边去寻找第 k - m - 1 大的数即可。
④如果 k < m + 1，说明第 k 大数在左边，我们在左边去寻找第 k 大的数即可。

原数组 3 7 6 1 5

如果把 5 作为分区点，那么数组最后就会变成下边的样子, i 指向最终的分区点
7 6 5 1 3
    ^
    i

public int findKthLargest(int[] nums, int k) {
    return findKthLargestHelper(nums, 0, nums.length - 1, k);
}

private int findKthLargestHelper(int[] nums, int start, int end, int k) {
    int i = start;
    int pivot = nums[end];//分区点
    //将 i 的左半部分存比分区点大的数
    //将 i 的右半部分存比分区点小的数
    for (int j = start; j < end; j++) {
        if (nums[j] >= pivot) {
            int temp = nums[i];
            nums[i] = nums[j];
            nums[j] = temp;
            i++;
        }
    }
    //分区点放到 i 的位置
    int temp = nums[i];
    nums[i] = pivot;
    nums[end] = temp;
    //左边的数量加上 1
    int count = i - start + 1;
    if (count == k) {
        return nums[i];
    //从右边去继续寻找
    } else if (count < k) {
        return findKthLargestHelper(nums, i + 1, end, k - count);
    //从左边去继续寻找    
    } else {
        return findKthLargestHelper(nums, start, i - 1, k);
    }
}

解法三 最大堆，优先队列

使用优先队列，建一个最大堆，然后依次弹出元素，弹出的第 k 个元素就是我们要找的。

这里直接使用 java 提供的优先队列

public int findKthLargest(int[] nums, int k) {
    Comparator<Integer> cmp;
    //java 默认的是建最小堆，所以我们使用比较器来改变优先级。
    cmp = new Comparator<Integer>() {
        @Override
        public int compare(Integer i1, Integer i2) {
            // TODO Auto-generated method stub
            return i2 - i1;
        }
    };

    // 建立最大堆
    Queue<Integer> q = new PriorityQueue<Integer>(cmp);
    for (int i = 0; i < nums.length; i++) {
        q.offer(nums[i]);
    }

    for (int i = 0; i < k - 1; i++) {
        q.poll();
    }
    return q.poll();
}