携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第20天,点击查看活动详情
概述
数组算是一个并不复杂的数据结构,但就是这个,面试官就能在面试过程中考察出很多东西来。例如经典的排序问题,二分搜索,滑动窗口等等,都是在数组这种最基础等数据结构中处理问题的。
我们做数组类算法问题的时候,常常需要定义一个变量,明确该变量的意义,并不停的维护这个变量,也需要特别注意初始值
和边界
问题。
题目
给定整数数组 nums
和整数 k
,请返回数组中第 k
个最大的元素。
请注意,你需要找的是数组排序后的第 k
个最大的元素,而不是第 k
个不同的元素。
你必须设计并实现时间复杂度为 O(n)
的算法解决此问题。
示例
输入: [3,2,1,5,6,4], k = 2
输出: 5
输入: [3,2,3,1,2,4,5,5,6], k = 4
输出: 4
提示
1 <= k <= nums.length <= 10^5
-10^4 <= nums[i] <= 10^4
代码实现
排序
class Solution {
public int findKthLargest(int[] nums, int k) {
// 排序后返回
int len = nums.length;
Arrays.sort(nums);
return nums[len - k];
}
}
优先队列
class Solution {
public int findKthLargest(int[] nums, int k) {
int len = nums.length;
// 优先队列排序
PriorityQueue<Integer> minHeap = new PriorityQueue<>(k, Comparator.comparingInt(a -> a));
// 入队操作
for (int i = 0; i < k; i++) {
minHeap.offer(nums[i]);
}
for (int i = k; i < len; i++) {
Integer topElement = minHeap.peek();
// 反转操作
if (nums[i] > topElement) {
minHeap.poll();
minHeap.offer(nums[i]);
}
}
return minHeap.peek();
}
}