记录新手小白的做题过程。
题目:
给你一个整数数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。
返回 滑动窗口中的最大值 。
示例 1:
输入:nums = [1,3,-1,-3,5,3,6,7], k = 3
输出:[3,3,5,5,6,7]
解释:
滑动窗口的位置 最大值
--------------- -----
[1 3 -1] -3 5 3 6 7 3
1 [3 -1 -3] 5 3 6 7 3
1 3 [-1 -3 5] 3 6 7 5
1 3 -1 [-3 5 3] 6 7 5
1 3 -1 -3 [5 3 6] 7 6
1 3 -1 -3 5 [3 6 7] 7
示例 2:输入:nums = [1], k = 1
输出:[1]
天哪,标签是困难欸,我还没看看题就怂了。我记得滑动窗口似乎是个不那么简单的算法。
思路:嗯,看了之后感觉也还好。思路大概是用队列来存储数据,最先往队列里面放k个数据,找到他们的最大值,然后取出队列第一个数,再往队尾压入一个元素,若取出的元素是之前的最大值,就重新遍历滑动窗口的k个元素,找最大值,若取出的元素不是最大值,那就直接将压入的元素与之前的最大值比较,获得这次的最大值就可以了。
感觉思路还挺清晰的,不算很难,代码如下:
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
vector<int> ans;
queue<int> stl;
int i=0;int max=INT_MIN;
while(i<k){//会不会有可能k>nums.size()?题目提示说不可能,放心干,先放入k个元素在队列中
stl.push(nums[i]);
i++;
if(nums[i]>max)//找第一个滑动窗口的最大值
max=nums[i];
}
ans.push_back(max);//放进第一个值
for(;i<nums.size();i++){
if(max==stl.front()){//如果上一次的最大值是队头,那么要重新比较k内元素
stl.pop();
stl.push(nums[i]);
max=INT_MIN;
for(int j=i-k+1;j<=i;j++){
if(nums[j]>max)
max=nums[j];
}
}
else{
if(max<nums[i])
max=nums[i];
}
ans.push_back(max);
}
return ans;
}
};
害,可是不知道为什么执行错误,一个都没通过。
官方代码:
方法一:优先队列
对于「最大值」,我们可以想到一种非常合适的数据结构,那就是优先队列(堆),其中的大根堆可以帮助我们实时维护一系列元素中的最大值。
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
int n = nums.size();
priority_queue<pair<int, int>> q;
for (int i = 0; i < k; ++i) {
q.emplace(nums[i], i);//放入前k个元素
}
vector<int> ans = {q.top().first};//最大堆的堆顶是最大元素
for (int i = k; i < n; ++i) {
q.emplace(nums[i], i);
while (q.top().second <= i - k) {//判断数字是否已经在滑动窗口外
q.pop();//弹出
}
ans.push_back(q.top().first);//取最大堆的数字
}
return ans;
}
};
收获:1.优先队列的一点点知识:
点击链接 emplace()原地构造一个数据并插入,
2.在弹出元素时要用的是while,而不是if,因为这里并不是插入一个删除一个,堆中也许还有很多在窗口之外,但一直在堆顶的元素,要全部取出才能到符合滑动窗口内
方法二:单调队列
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
int n = nums.size();
deque<int> q;
for (int i = 0; i < k; ++i) {
while (!q.empty() && nums[i] >= nums[q.back()]) {
q.pop_back();
}
q.push_back(i);
}
vector<int> ans = {nums[q.front()]};
for (int i = k; i < n; ++i) {
while (!q.empty() && nums[i] >= nums[q.back()]) {//每次插入数据都要遵循递减
q.pop_back();
}
q.push_back(i);
while (q.front() <= i - k) {//还要判断一下最大值是否在窗口内//注意这里能否取等号
q.pop_front();
}
ans.push_back(nums[q.front()]);
}
return ans;
}
};
这个方法的思路就是要建成一个单调递减的双端队列,将队顶的数值永远维护在滑动窗口里的最大值。即每取一个数与队列尾部比较,若这个数大于尾部,则删除尾部,与新的尾部继续比较,直到找到一个能大于这个数的尾部或队列已经为空,再插入元素。
因为要通过下标计算元素是否在窗口内,官方的存在队列里的不是元素,而是下标。
要注意双端队列的操作函数写法。
大概思路已经懂得,记得复盘!
下一题!