【字节专题】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 ≤ 数组的长度。

方法一

可以开一个k大小的最小堆（时间复杂度为nlogk），或者n-k大小的最大堆（时间复杂度为nlog(n-k)），堆的做法选哪个就看k的大小。

public int findKthLargest(int[] nums, int k) {
    PriorityQueue<Integer> minHeap = new PriorityQueue<>();
    for (int i = 0;i<nums.length;i++){
        minHeap.add(nums[i]);
        if (minHeap.size()>k)
            minHeap.poll();
    }
    return minHeap.peek();
}

方法二

求数组中第k大可以使用快排算法思想做，每次使用partition函数返回的p和k比较，缩小范围，当p==k，就是找到了第k大。（每次partition函数就会确定下标为p的元素的最终位置）

public int findKthLargest(int[] nums, int k) {
    int l = 0, r = nums.length-1;
    k = nums.length - k;
    while (l<=r){
        int p = partition(nums, l, r);
        if(p==k)
            return nums[k];
        else if(k<p){
            r = p - 1;
        }else
            l = p + 1;

    }
    return 0;
}

private int partition(int[] arr, int l, int r){
    int v = arr[l];

    int i = l+1, j = r;
    while (i<=j){
        while (i<=j && arr[i]<v) i++;
        while (i<=j && arr[j]>v) j--;
        if(i>j)
            break;
        swap(arr, i, j);
        i++;
        j--;
    }
    swap(arr, l, j);
    return j;
}
private void swap(int[] arr, int i, int j){
    int t = arr[i];
    arr[i]=arr[j];
    arr[j]=t;
}