题目
有一个整型数组arr和一个大小为w的窗口从数组的最左边滑到最右边,窗口每次向右边滑动一个位置。
- 输入:整形数组arr,窗口大小为w。
- 输出:一个长度为n-w+1的数组res,res[i]表示每一种窗口状态下的最大值。
- 要求:时间复杂度为O(n)。
思路
跟剑指offer是同一题目,采用双端队列,在队列中存放原数组的下标,分别在队列头和队列尾进行对应操作。
判断条件能够考虑清楚就好具体代码及过程请参考原书
代码
#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;
void maxNumInDeque(int* in, int* &out, int len, int k)
{
deque<int> dq;
int i;
for (i = 0; i < k; i++)
{
while (!dq.empty() && in[i] > in[dq.back()])
{
dq.pop_back();
}
dq.push_back(i);
}
for (int j = k; j < len; j++)
{
out[j - k] = in[dq.front()];
while (!dq.empty() && in[j] > in[dq.back()])
dq.pop_back();
while (!dq.empty() && j - dq.front() >= k)
dq.pop_front();
dq.push_back(j);
}
out[len - k] = dq.front();
}
int main()
{
int len = 8;
int k = 3;
int input[] = { 4, 3, 5, 4, 3, 3, 6, 7 };
int* out = new int[len - k + 1];
maxNumInDeque(input, out, len, k);
for (int i = 0; i < len - k + 1; i++)
{
cout << out[i] << endl;
}
getchar();
return 0;
}