题目描述
给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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>res;
if(size<1)
return res;
int lenth = num.size();
for (int i = 0; i<lenth - size + 1; i++)
{
int max= num[i];
for (int j = i; j<i + size; ++j)
{
max = num[j]>max ? num[j] : max;
}
res.push_back(max);
}
return res;
}
};
使用 一个双端队列 deque ,始终让队头是窗口内最大的元素的下标,如果在遇到比队头大的元素让队出空 ,压入这个大元素。
如果遇到比队头大元素小的,也为了防止之后头在窗口划走的时候出队, 这个小元素成为 最大的,所以也要 从队尾入队保存下来。 都是保存元素下标!!!
#include <deque>
class Solution {
public:
vector<int> maxInWindows(const vector<int>& num, unsigned int size)
{
vector<int> res;
deque<int> q;
int lenth = num.size();
q.push_back(0);
for (int i = 1; i<lenth; ++i)
{
if (q.front() + size == i)//当前最大的 是否还在窗口内
{
q.pop_front();
deque<int> ::iterator it = q.begin()+1 ;
int max_index=q.front();
while (it != q.end())
{
if (num[*it]>num[max_index])
max_index = *it;
++it;
}
q.clear();
q.push_back(max_index);
/*
int max, m_max;
do{
q.pop_front();
deque<int> ::iterator it = q.begin() + 1;
if (it == q.end())
break;
max = num[q.front()];
m_max = num[*it];
} while (max<m_max);
*/
}
else if (num[i]>num[q.front()]) //是否有更大的元素出现
{
q.clear();
q.push_back(i);
}
else{//最大的出队后 现在的元素可能成为最大的
q.push_back(i);
}
if ((i + 1) >= size)
{
res.push_back(num[q.front()]);
}
}
return res;
}
};