问题描述:整型数组array和width的窗口从数组最左边滑到最右边,每次向右边滑动一个位置,返回一个长度为size-width+1的数组res,res[i]表示第种窗口状态下的最大值。
实例:
输入:
vector<int>array={4,3,5,4,3,3,6,7};
int width=3;
输出:
vector<int> res = {5,5,5,4,6,7};
解题思路:时间复杂度为o(n)
利用队列q来实现窗口最大值的更新,存储数组array中的下标
遍历到array的第i个元素,q的放入规则:
1、如果q为空,则将下标放入,结束
2、如果q不为空,取出当前队尾下标j,进行比较:如果array[i]< array[j],则将i放入,结束;
如果array[i]> array[j],则循环q出队列,直到array[i]<array[j]或者队列为空
如果对头下标 = i-width,说明对头过期,出队列(先进行判断)
代码实现:
#include<iostream>
#include<queue>
#include<vector>
#include<string>
using namespace std;
int main()
{
vector<int>nums = { 4, 3, 5, 1, 3, 3, 6, 7 };
int width = 3;
vector<int>res;
queue<int> max_index;
for (int i = 0; i < nums.size(); i++){
//注: 个人觉得应该先进行判断下标是否过期,在进行两个数字大小的比较
if (!max_index.empty()&&max_index.front() == i - width)
max_index.pop();
while (!max_index.empty() && nums[max_index.front()] <= nums[i])
max_index.pop();
max_index.push(i);
if (i >= width - 1)//里面存储的是最大值
res.push_back(nums[max_index.front()]);
}
for (int i = 0; i < res.size(); i++)
cout << res[i] << endl;
system("pause");
return 0;
}
参考:左程云编著的《程序员代码面试指南》的第19页
https://www.cnblogs.com/haozhengfei/p/a14049ec0869a8125a69f3af37471c77.html