剑指 Offer 59 - I. 滑动窗口的最大值https://leetcode.cn/problems/hua-dong-chuang-kou-de-zui-da-zhi-lcof/
先设计一个数据结构:单调队列
使其实现:向其 push() 时 自动把队尾小于插入值的元素 弹出,再 push
实际存储的即为 一串单调递减的元素
于是,当我们将其用于 求滑动窗口最大值时,可以通过单调队列存放滑动窗口内可能的最大值 来实现:
- 每次取最大值 就是直接取单调队列的队头元素;
- 每次扩大窗口 就是直接往单调队列push;
- 每次收缩窗口 就是看离开窗口的元素是否为单调队列的队头元素、若是则pop
单调队列的实现
// 单调队列,存放的元素单调递减
class MonotonicQueue
{
public:
void push(const int val)
{
// 先把尾部小于val的元素弹出
while (!d.empty() && d.back() < val)
{
d.pop_back();
}
d.push_back(val);
}
void pop()
{
if (!d.empty())
{
d.pop_front();
}
}
int max()
{
return d.empty() ? -1 : d.front();
}
private:
deque<int> d;
};
解法代码
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
int n = nums.size();
MonotonicQueue mq;
vector<int> res;
for (int i = 0; i < n; ++i)
{
mq.push(nums[i]); // 滑动窗口增长
// 前 k - 1 个元素直接入队
if (i < k - 1)
{
continue;
}
res.push_back(mq.max());
if (nums[i - k + 1] == mq.max()) // 滑动窗口收缩
{
mq.pop();
}
}
// base case
if (res.empty())
{
res.push_back(mq.max());
}
return res;
}