题目链接:239. 滑动窗口最大值 - 力扣(LeetCode)
一个滑动窗口一直滑,记录下每次滑的最大值
滑动窗口
这道题和之前的滑动窗口的题目有点类似,要避免直接两层循环暴力求解,可以使用滑动窗口【LeetCode热题100】【滑动窗口】找到字符串中所有字母异位词_找到字符串中所有字母异位 题解-CSDN博客
要寻找这个滑动窗口的最大值最快的方法是使用一个大顶堆,堆的插入元素的时间复杂度为logn,这样不用遍历窗口的每个元素就可以找出最大值
但这样还有一个问题,那就是滑动窗口移动的时候,如果删除左边被移出窗口的元素,堆删除指定元素并不简单,解决的方法就是不删除,当堆顶元素为已经移出窗口的元素时,pop堆顶元素就行,这样就可以避免找到的最大值是已经移除的元素
为了实现判断这个元素是否已经移除窗口,我们采用二元组来存储每个元素本身和它的索引,当索引小于等于当前的i-k,说明这个元素已经不在了,做掉
还有一点就是C++没有堆这个容器,但是有优先队列,这个是堆实现的可以当成堆来用,而且默认是大顶堆,而C++中的二元组pari比较大小的时候默认是先比较第一个,所以我们需要把元素本身放第一个
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
vector<int>ans;
priority_queue<pair<int,int>>heap;
for(int i=0;i<k;++i)
heap.emplace(nums[i],i);
ans.push_back(heap.top().first);
for(int i=k;i<nums.size();++i){
heap.emplace(nums[i],i);
while(heap.top().second<=i-k)
heap.pop();
ans.push_back(heap.top().first);
}
return ans;
}
};
rust
use std::collections::BinaryHeap;
impl Solution {
pub fn max_sliding_window(nums: Vec<i32>, k: i32) -> Vec<i32> {
let k = k as usize;
let mut ans: Vec<i32> = Vec::new();
let mut maxheap: BinaryHeap<(i32, usize)> = BinaryHeap::new();
for i in 0..k {
maxheap.push((nums[i], i));
}
if let Some(&x) = maxheap.peek() {
ans.push(x.0);
}
for i in k..nums.len() {
loop {
if let Some(&x) = maxheap.peek() {
if x.1 <= i - k {
maxheap.pop();
} else {
break;
}
}else{
break;
}
}
maxheap.push((nums[i], i));
if let Some(&x) = maxheap.peek() {
ans.push(x.0);
}
}
ans
}
}