题目如下
思路与代码
对于数组,假设我们当前遍历到下标i,对于下标i+1的元素(假设i和i+1都在同一个窗口),如果比arr[i]大,说明了什么?
如果arr[i+1] 已经大于了 arr[i], 那么还要arr[i]有什么用.就有点“既生瑜何生亮”的感觉。
如果arr[i+1] < arr[i]呢?显然arr[i]还是需要保留的。为什么呢?
因为又可以arr[i] 对于下一个arr[i+1]所在的窗口来说,arr[i]已经失效了。
假设这里有那么一个容器可以保留上述操作。
遍历数组的每一个元素,
如果容器为空,则直接将当前元素加入到容器中。
如果容器不为空,则让当前元素和容器的最后一个元素比较,如果大于,则将容器的最后一个元素删除,然后继续讲当前元素和容器的最后一个元素比较
如果当前元素小于容器的最后一个元素,则直接将当前元素加入到容器的末尾
如果容器头部的元素已经不属于当前窗口的边界,则应该将头部元素删除
总结一下,首先容器中放的元素应该是单调递减的。然后还有删除容器头部元素和最后一个元素的操作。因此,这样的数据结构就是双端队列。c++中就是deque
看到为什么i+1>size开始push_back到ret里面就很容易明白了!
class Solution {
public:
vector<int> maxInWindows(const vector<int>& num, unsigned int size) {
vector<int> ret;
if(num.size()==0||size<1||num.size()<size) return ret;
int n=num.size();
deque<int> dq;
for(int i=0;i<n;i++){
while(!dq.empty()&&num[dq.back()]<num[i]){
dq.pop_back();
}
dq.push_back(i);
if(dq.front()+size<=i)
dq.pop_front();
if(i+1>=size)
ret.push_back(num[dq.front()]);
}
return ret;
}
};