题目:给定一个数组和滑动窗口的大小,请找出所有滑动窗口里的最大值。举例说明,例如,如果输入数组{2,3,4,2,6,2,5,1}及滑动窗口的大小为3,那么一共存在6个滑动窗口,它们的最大值分别为{4,4,6,6,6,5}。
思路:
用一个双端队列来存储窗口中值的下标,保证队首对应的元素为队列中最大值。
当插入元素小于队尾对应元素,加在后面,因为等前面值过了它可能就是最大的了。
当插入元素大于队尾对应元素,那么在这个窗口中,前面那个小的值就没什么意义了,因为我最后一个值在的话,你肯定不是最大值,所以把前面值一直pop掉。
代码:
#include <iostream>
#include <vector>
#include<deque>
#include<algorithm>
using namespace std;
vector<int> maxInWindow(vector<int> input,int win_len){
vector<int> res;
deque<int> index;//记录是下标,而不是具体元素值,这样可以知道队列长度。
//先把窗口长度的队列装满,然后再一个一个添加
if (input.size() < win_len) return res;
for (int i = 0; i < win_len; i++) {
while (!index.empty() && input[index.back()] < input[i]) {
index.pop_back(); //如果队列中最后一个元素的值比待加入元素小,一直删除。因为有后面那个大的元素在,前面没什么用
}
index.push_back(i);
}
res.push_back(input[index.front()]); //先把前面那个窗口中最大的拿下,最大的肯定在队首。
for (int i = win_len; i < input.size(); i++) {
if (i - index.front() == win_len)index.pop_front(); //窗口长度限制
while (!index.empty() && input[index.back()] < input[i]) {
index.pop_back();
}
index.push_back(i);
res.push_back(input[index.front()]);
}
return res;
}
int main()
{
int A[] = {2,3,4,2,6,2,5,1};
vector<int> input;
for (int x : A) input.push_back(x);
int win_len = 3;
vector<int> res = maxInWindow(input,win_len);
for (auto x : res) cout << x << " ";
//4 4 6 6 6 5
return 0;
}
结果:
tips:vector中,input.front()可以取出第一个元素,input.back()可以取出最后一个元素。