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

Posted 2021-09-07 易小顺

算法记录

LeetCode 题目：

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

---

思路

• 算法记录
• 说明
• • 一、题目
  • 二、分析
• 总结

---

说明

一、题目

  输入: [3,2,1,5,6,4] 和 k = 2。输出: 5

二、分析

• 题解很清楚，输出排序后的第 K 大的元素，也就是说我们需要维护一个最大序列，取出第 K 个值即可。
• 排序方式很多，这里采用堆排序。

class Solution {
    public int findKthLargest(int[] nums, int k) {
        Queue<Integer> queue = new PriorityQueue<>(new Comparator<Integer>() {
            public int compare(Integer e1, Integer e2) {
                return e2 - e1;
            }
        }); 
        for(int i :nums) queue.add(i);
        while(--k != 0) {
            queue.poll();
        }
        return queue.poll();
    }
}

---

总结

熟悉数组遍历方法。