題目
給定一個數組和滑動窗口的大小,找出所有滑動窗口裏數值的最大值。例如,如果輸入數組{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]}
方法一:
1、分析
使用暴力的方法來進行,分別求出每個子數組中的最大值。時間複雜度爲 ,k爲滑動窗口的大小,n爲數組的大小。
2、代碼
class Solution {
public:
vector<int> maxInWindows(const vector<int>& num, unsigned int size)
{
vector<int> res;
unsigned int len=num.size();
if(len<=0 ||size==0)
return res;
unsigned int count=len-size+1;
for(unsigned int i=0;i<count;++i)
{
unsigned int end=i+size;
int maxNum=num[i];
for(unsigned int j=i+1;j<end;++j)
{
if(maxNum<num[j])
maxNum=num[j];
}
res.push_back(maxNum);
}
return res;
}
};
方法二
1、分析
使用STL中的隊列 deque 來將最大值的下標保存下來,同時保證該隊列的首元素就是最大值的下標。
2、代碼
class Solution {
public:
vector<int> maxInWindows(const vector<int>& num, unsigned int size)
{
vector<int> res;
if(num.size()>=size && size>=1)
{
deque<int> dq;
//求第一個滑窗中的size個元素的最大值
for(unsigned int i=0;i<size;++i)
{
while(!dq.empty() && num[i]>=num[dq.back()])
{
//pop_back之後,前面的元素就會變成末尾
dq.pop_back();
}
dq.push_back(i);
}
//接下來將後續添加的元素與dq中的進行比較,求最大值
for(unsigned int j=size;j<num.size();++j)
{
res.push_back(num[dq.front()]);
while(!dq.empty() && num[j]>=num[dq.back()])
dq.pop_back();
if(!dq.empty() && dq.front()<=j-size)
dq.pop_front();
dq.push_back(j);
}
res.push_back(num[dq.front()]);
}
return res;
}
};