滑动窗口的最大值
题目描述
给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{2,3,4,2,6,2,5,1}及滑动窗口的大小3,那么一共存在6个滑动窗口,他们的最大值分别为{4,4,6,6,6,5}; 针对数组{2,3,4,2,6,2,5,1}的滑动窗口有以下6个: {[2,3,4],2,6,2,5,1}, {2,[3,4,2],6,2,5,1}, {2,3,[4,2,6],2,5,1}, {2,3,4,[2,6,2],5,1}, {2,3,4,2,[6,2,5],1}, {2,3,4,2,6,[2,5,1]}。
使用一个队列做小标索引, 保存窗内最大值的下标, 还要判断这个索引是否在当前窗内
class Solution {
public:
vector<int> maxInWindows(const vector<int>& num, unsigned int size)
{
vector<int> ret;
if ((num.size() >= size ) && (size > 0)) {
deque<int> index; // 用于存放下标的索引
for (auto i = 0; i < size; i++) { // 存放前size个中的最大值
// 重复元素只存储一份, 所以用大于等于号
while ((! index.empty()) && num[index.back()] <= num[i])
index.pop_back();
index.push_back(i);
}
for (auto i = size; i < num.size(); i++) {
ret.push_back(num[index.front()]);
// 后插入的元素值比前面的元素大时, 这个元素就是是窗里最大元素, 要把前面元素都弹出
while ((!index.empty()) && num[index.back()] <= num[i])
index.pop_back();
// 当前头部索引不在窗范围内时要弹出头部索引
if ((!index.empty() && index.front() <= (i - size)))
index.pop_front();
// 若后插入的元素比头部索引小时, 由于头部索引会移除; 移出后, 后插入的元素可能会是最大值
index.push_back(i);
}
// 上一个fou循环中i指向的是窗的后一个元素, 当i=size()时,
//最后一个窗内最大值并没有插入ret中
ret.push_back(num[index.front()]);
}
return ret;
}
};
比较简陋, 笨方法, 一次下移滑动窗口, 然后找最大值
class Solution {
public:
int findMax(const vector<int>& vt, int pos, unsigned int num) {
int max = vt[pos];
for(auto i = 0; i < num; i++) {
if (vt[pos+i] > max)
max = vt[pos+i];
}
return max;
}
vector<int> maxInWindows(const vector<int>& num, unsigned int size)
{
vector<int> ret;
if (num.empty() || size < 1 || size > num.size())
return ret;
for (auto i = 0; i < (num.size() - size + 1); i++) {
int temp = 0;
temp = findMax(num, i, size);
ret.push_back(temp);
}
return ret;
}
};