题目一:滑动窗口的最大值
给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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、思路
我们可以使用一个双端队列deque。
我们可以用STL中的deque来实现,接下来我们以数组{2,3,4,2,6,2,5,1}为例,来细说整体思路。
数组的第一个数字是2,把它存入队列中。第二个数字是3,比2大,所以2不可能是滑动窗口中的最大值,因此把2从队列里删除,再把3存入队列中。第三个数字是4,比3大,同样的删3存4。此时滑动窗口中已经有3个数字,而它的最大值4位于队列的头部。
第四个数字2比4小,但是当4滑出之后它还是有可能成为最大值的,所以我们把2存入队列的尾部。下一个数字是6,比4和2都大,删4和2,存6。就这样依次进行,最大值永远位于队列的头部。
但是我们怎样判断滑动窗口是否包括一个数字?应该在队列里存入数字在数组里的下标,而不是数值。当一个数字的下标与当前处理的数字的下标之差大于或者相等于滑动窗口大小时,这个数字已经从窗口中滑出,可以从队列中删除。
整体过程示意图:
实现代码如下:
class Solution {
public:
vector<int> maxInWindows(const vector<int>& num, unsigned int size)
{
vector<int> res;
if(num.size() < size || size < 1)
return res;
deque<int> index;
for(int i=0;i<size;++i){
while(!index.empty() && num[index.back()]<=num[i])
index.pop_back();
index.push_back(i);
}
for(int i=size;i<num.size();++i){
res.push_back(num[index.front()]);
while(!index.empty() && num[index.back()]<=num[i])
index.pop_back();
if(!index.empty() && i-index.front()>=size)
index.pop_front();
index.push_back(i);
}
res.push_back(num[index.front()]);
return res;
}
};
面试题59(二):队列的最大值 | |
// 题目:给定一个数组和滑动窗口的大小,请找出所有滑动窗口里的最大值。例如, | |
// 如果输入数组{2, 3, 4, 2, 6, 2, 5, 1}及滑动窗口的大小3,那么一共存在6个 | |
// 滑动窗口,它们的最大值分别为{4, 4, 6, 6, 6, 5}, |
解析:和上题思路相似,用双端队列去解决
#include <iostream>
#include <cstdio>
#include <deque>
using namespace std;
class QueueWithMax{
public:
void push_back(int val){
while(!maxinums.empty() && val>=maxinums.back().num)//注意是back
maxinums.pop_back();
node temp = node{index,val};
data.push_back(temp);
maxinums.push_back(temp);
++index;
}
void pop_front(){
if(data.empty())
return ;
if(maxinums.front().index == data.front().index)
maxinums.pop_front();
data.pop_front();
}
int max(){
if(data.empty())
return -1;
return maxinums.front().num;
}
private:
struct node{
int index;
int num;
};
deque<node> data;
deque<node> maxinums;
int index = 0;
};
void Test(const char* testName, QueueWithMax& queue, int expected)
{
if(testName != nullptr)
printf("%s begins: ", testName);
if(queue.max() == expected)
printf("Passed.\n");
else
printf("FAILED.\n");
}
int main(){
QueueWithMax queue;
// {2}
queue.push_back(2);
Test("Test1", queue, 2);
// {2, 3}
queue.push_back(3);
Test("Test2", queue, 3);
// {2, 3, 4}
queue.push_back(4);
Test("Test3", queue, 4);
// {2, 3, 4, 2}
queue.push_back(2);
Test("Test4", queue, 4);
// {3, 4, 2}
queue.pop_front();
Test("Test5", queue, 4);
// {4, 2}
queue.pop_front();
Test("Test6", queue, 4);
// {2}
queue.pop_front();
Test("Test7", queue, 2);
// {2, 6}
queue.push_back(6);
Test("Test8", queue, 6);
// {2, 6, 2}
queue.push_back(2);
Test("Test9", queue, 6);
// {2, 6, 2, 5}
queue.push_back(5);
Test("Test9", queue, 6);
// {6, 2, 5}
queue.pop_front();
Test("Test10", queue, 6);
// {2, 5}
queue.pop_front();
Test("Test11", queue, 5);
// {5}
queue.pop_front();
Test("Test12", queue, 5);
// {5, 1}
queue.push_back(1);
Test("Test13", queue, 5);
return 0;
}