用Java写下面题目:给定整数数组 nums 和整数 k,请返回数组中第 k 个最大的元素。 请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。 你必须设计并实现时间复杂度为 O(n) 的算法解决此问题。
时间: 2023-12-28 22:03:24 浏览: 82
寻找数组中第k大的元素
好的,以下是使用Java解决该问题的代码:
```
public int findKthLargest(int[] nums, int k) {
PriorityQueue<Integer> heap = new PriorityQueue<>();
for (int num : nums) {
heap.offer(num);
if (heap.size() > k) {
heap.poll();
}
}
return heap.peek();
}
```
该算法使用了一个大小为k的最小堆来维护数组中前k大的元素。首先将数组中的元素加入堆中,如果堆的大小超过了k,则将堆顶元素弹出。最终堆顶元素即为第k个最大的元素。
由于最小堆的大小为k,所以堆中的元素个数最多为k个。因此,时间复杂度为O(nlogk)。然而,由于k是常数,所以该算法的时间复杂度为O(n)。
注意,在使用PriorityQueue时,需要将其初始化为一个最小堆,因此代码中使用了默认的构造函数。如果要将其初始化为最大堆,则需要将构造函数改为`PriorityQueue<Integer>(Collections.reverseOrder())`。
阅读全文