【leetcode每日刷题】215. Kth Largest Element in an Array

找第k大的数

1、使用最小堆，当大于k的时候并且插入的值大于堆顶元素的时候，删除堆顶元素，在把新元素加入堆中。

public int heap(int[] nums, int k) {
        PriorityQueue<Integer> heap = new PriorityQueue<>();
        for(int num:nums){
            if(heap.size() < k) heap.offer(num);
            else{
                if(num > heap.peek()){
                    heap.poll();
                    heap.offer(num);
                }
            }
        }
        return heap.peek();
    }

2、使用快排的方法

class Solution {
    public int findKthLargest(int[] nums, int k) {
       int id = partition(nums, 0, nums.length-1, nums.length-k);
        return nums[id];
    }
    public int partition(int[] nums, int left, int right, int k){
        int i = left - 1;
        int pivot = nums[right];
        for(int j=left; j<=right; j++){
            if(nums[j]<pivot){
                i++;
                int temp = nums[i];
                nums[i] = nums[j];
                nums[j] = temp;
            }
        }
        i++;
        int temp = nums[i];
        nums[i] = nums[right];
        nums[right] = temp;
        if(i==k) return i;
        else if(i<k) return partition(nums, i+1, right, k);
        else return partition(nums, left, right-1, k);
    }
}