239. 滑动窗口最大值 (一刷至少需要理解思路)
之前讲的都是栈的应用,这次该是队列的应用了。
本题算比较有难度的,需要自己去构造单调队列,建议先看视频来理解。
题目链接/文章讲解/视频讲解:代码随想录
思路
public static int[] maxSlidingWindow(int[] nums, int k) {
int len= nums.length-k+1;
int[] res = new int[len];
int num=0;
MyQueue myQueue = new MyQueue();
// 先把前k个元素放入队列
for (int i = 0; i < k; i++) {
myQueue.push(nums[i]);
}
res[num++] = myQueue.peek();
for (int i = k; i < nums.length; i++) {
// 滑动窗口移除最前面的元素,移除是判断该元素是否放入队列
myQueue.pop(nums[i-k]);
// 滑动窗口加入最后面的元素
myQueue.push(nums[i]);
res[num++]=myQueue.peek();
}
return res;
}
class MyQueue{
Deque<Integer> deque =new LinkedList<>();
void pop(int val){
if(!deque.isEmpty() && deque.peek()==val){
deque.pop();
}
}
void push(int val){
while (!deque.isEmpty() && val>deque.getLast()){
deque.removeLast();
}
deque.add(val);
}
int peek(){
return deque.peek();
}
}
347.前k个高频元素
思路
// 优先级队列:对外接口只是从队头取元素,从队尾添加元素。
// 优先级队列内部元素是自动依照元素的权值排列
public static int[] topKFrequent(int[] nums, int k) {
// 统计元素出现的频率,用map来进行统计
Map<Integer, Integer> map = new HashMap<>();
for (int num:nums){
map.put(num,map.getOrDefault(num,0)+1);
}
// 对频率排序,优先级队列
// 创建优先级队列,其中存储的元素是整形数组int[]。通过传入一个比较器作为参数,定义了优先队列的排序规则。
PriorityQueue<int[]> pq = new PriorityQueue<>((pair1,pair2)->pair1[1]-pair2[1]);
for (Map.Entry<Integer,Integer> entry:map.entrySet()){
if(pq.size()<k){
pq.add(new int[]{entry.getKey(),entry.getValue()});
}else {
// 进行索引操作,获取数组的第二个元素
if (entry.getValue()>pq.peek()[1]){
pq.poll();
pq.add(new int[]{entry.getKey(),entry.getValue()});
}
}
}
// 我们用小顶堆,因为要统计最大前k个元素,只有小顶堆每次将最小的元素弹出,最后小顶堆积累的才是前k个最大元素
int[] ans = new int[k];
for(int i=k-1;i>=0;i--){//依次弹出小顶堆,先弹出的是堆的根,出现次数少,后面弹出的出现次数多
ans[i] = pq.poll()[0];
}
return ans;
}