滑动窗口(数组中一定长度范围内的最大值)

问题描述:整型数组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

猜你喜欢

转载自blog.csdn.net/liugg2016/article/details/81975039