栈与队列 (困难)
Q1 I. 滑动窗口的最大值
给定一个数组 nums
和滑动窗口的大小 k
,请找出所有滑动窗口里的最大值。
思路:
本题的关键是如何计算每次窗口右移后的最大值。如果每次都通过遍历获取,那么总时间复杂度O(n* k )当n与k相近时达到n^2;因此我们需要找到使计算窗口最大值时间复杂度更低的方法。
我们在窗口右移同时维护一个单调队列,队列内次序从大到小。
每当窗口右移时:
-
先判断此时最大元素是否是最右侧的元素,若是,则需要pop_front从队首出队
-
在新元素入队前,先使队列内小于新元素的元素出队列,由此我们可以保证队列的单调性。
-
即使我们当前右移删除的的最左元素是最大元素,由于单调队列的特性,原窗口小于最左元素的第二大值也一定在队列内
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
int len = nums.size();
if (!len) return{};
if(k == 1) return nums;
vector<int> res;
deque<int> deq;
int right = 0;
// 形成第一个大小为k的窗口
while (right < k) {
//保证队列单调性,即每个元素进队列前,保证在他之前的元素都比他大
//此时即使窗口右移,也能保证在删掉最左侧元素后找出原窗口剩余元素的最大值
while (!deq.empty() && deq.back() < nums[right])
deq.pop_back();
deq.push_back(nums[right]);
++right;
}
res.push_back(deq.front());
// 向后滑动窗口
while (right < len) {
// 判断队首元素deq.front()是否为 移除窗口元素nums[right - k] (保证队列deq内只包含窗口内的元素)
if (deq.front() == nums[right - k]) deq.pop_front();
// 保持队列的单调性
while (!deq.empty() && deq.back() < nums[right])
deq.pop_back();
deq.push_back(nums[right]);
res.push_back(deq.front());
++right;
}
return res;
}
};
Q2 队列的最大值
请定义一个队列并实现函数 max_value 得到队列里的最大值,要求函数max_value、push_back 和 pop_front 的均摊时间复杂度都是O(1)。若队列为空,pop_front 和 max_value 需要返回 -1.
与上题类似,需要维护单调队列保存最大值。
class MaxQueue {
public:
queue<int> Q;
deque<int> dQ;
MaxQueue() { }
int max_value() {
if(Q.empty()) return -1;
return dQ.front();
}
void push_back(int value) {
Q.push(value);
while(!dQ.empty()&&dQ.back()<value) dQ.pop_back();
dQ.push_back(value);
}
int pop_front() {
int x = -1;
if(Q.empty()) return x;
else
{ x = Q.front();
Q.pop();
if(x == dQ.front()) dQ.pop_front();
}
return x ;
}
};
/**
* Your MaxQueue object will be instantiated and called as such:
* MaxQueue* obj = new MaxQueue();
* int param_1 = obj->max_value();
* obj->push_back(value);
* int param_3 = obj->pop_front();
*/