215. 数组中的第K个最大元素

一、题目

  给定整数数组 nums 和整数 k，请返回数组中第 k 个最大的元素。
  请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。
​

点击查看原题
难度级别： 中等

二、思路

1）优先队列

  建立一个优先队列，小根堆；当队列容量小于k时，直接插入；当容量等于k时，如果元素大于堆顶元素，那么取出堆顶元素，插入新元素

三、代码

1）优先队列

class Solution {
    public int findKthLargest(int[] nums, int k) {
        PriorityQueue<Integer> q = new PriorityQueue(k, new Comparator<Integer>(){
            @Override
            public int compare(Integer o1, Integer o2){return o1 - o2;}
        });
        for (int num : nums) {
            if (q.size() < k) {
                q.add(num);
            } else {
                if (num > q.peek()) {
                    q.poll();
                    q.add(num);
                }
            }
        }
        return q.peek();
    }
}

  时间复杂度为O(n*logk)，空间复杂度为O(k)