数组中的第K个最大元素

题目要求

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

暴力法

• 将数组从大到小排序，返回第 k - 1 项元素即为所求

public int findKthLargest(int[] nums, int k) {
        Integer[] integers = Arrays.stream(nums).boxed().toArray(Integer[]::new);
        Arrays.sort(integers, new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return o2 - o1;
            }
        });
        return integers[k - 1];
    }

堆

• 维护一个元素数目为k的一个小根堆，将所有元素入堆弹出堆，遍历完所有元素以后弹出堆顶元素即为所求。

public int findKthLargest(int[] nums, int k) {

        PriorityQueue<Integer> heap = new PriorityQueue<>();

        for (int num : nums) {
            heap.add(num);
            if (heap.size() > k) {
                heap.poll();
            }
        }

        return heap.poll();
    }