class Solution {
private:
class MyQueue {
public:
deque<int> que;
void pop(int val) {
if (!que.empty() && val == que.front()) {
que.pop_front();
}
}
void push(int val) {
while (!que.empty() && val > que.back()) {
que.pop_back();
}
que.push_back(val);
}
int front() {
return que.front();
}
};
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
MyQueue que;
for (int i = 0; i < k; i++) {
que.push(nums[i]);
}
vector<int> res;
res.push_back(que.front());
for (int i = k; i < nums.size(); i++) {
que.pop(nums[i - k]);
que.push(nums[i]);
res.push_back(que.front());
}
return res;
}
};
时间复杂度O(N)
思路:这题主要需要解决的就是滑动窗口的最大值,但看到是一个滑动窗口,可以想到它的实现类似一个队列,滑动窗口滑动的过程就是这个队列在pop和push,那我们到底怎么去求最大值呢,如果是暴力解法呢,就是遍历这个队列一遍,但是这时时间复杂度就不是线性的了,成为O(M*N),那如果要缩小时间复杂度,我们就会用到单调队列,当然单调队列是需要自己定义的,因为它是单调的,所以它既可以是单调递增也可以是单调递减,看自己需求。
定义单调队列时用到deque,是因为他可以对队列两边进行增加和删除还有查询两端元素。
C++中deque是stack和queue默认的底层实现容器,deque是可以两边扩展的,而且deque里元素并不是严格的连续分布的。
对于这题,因为我们每次添加元素都是从队尾添加元素,所以我们需要一个单调递减的队列,这样每次添加一个元素就和队列的队尾进行比较,如果比队尾大,就把队尾给pop_back掉,然后在和新的队尾比较,以此循环直至队列为空,然后添加该元素,这样就能保证队列的队首是最大的元素。
至于删除滑动窗口的第一个元素,其实就是判断删除的滑动窗口的第一个元素是不是我们队列里的队首,如果是就删,不是就不删。有可能不是队首的原因是滑动窗口的第一个元素在滑动窗口最大元素的前面,因为我定义的单调队列是递减的,每次push一个元素都会把队列中比这个元素小的元素给pop_back掉。