滑动窗口求解最大值

题目

给定一个长度为 n 的数组 num 和滑动窗口的大小 size ，找出所有滑动窗口里数值的最大值。

例如，如果输入数组{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]}。

窗口大于数组长度或窗口长度为0的时候，返回空。

数据范围： 1≤n≤100001≤n≤10000，0≤size≤100000≤size≤10000，数组中每个元素的值满足 ∣val∣≤10000∣val∣≤10000
要求：空间复杂度 O(n)O(n)，时间复杂度 O(n)O(n)

程序代码：

#include <iostream>
#include <deque>
#include <vector>
using namespace std;

/**
* @brief 滑动窗口中最大值
* @param num1 数据
* @param num2 窗口大小
* @return 每个窗口中最大值
*/
vector<int> sliding_window(vector<int>& number, int size)
{
	vector<int> max_result;//结果
	deque<int> dq;//用于记录索引值
	if (size<=number.size()&&size!=0)
	{
		//求出第一个窗口中的最大值
		for (int i = 0; i < size; i++) {
			//当dq不为空且新数比已存的末尾数大时，将末尾数删除，如果小则保存新数的索引
			while (!dq.empty() && number[dq.back()] < number[i]) {
				dq.pop_back();
			}
			dq.push_back(i);
		}
		//此时队列中第一个及为最大值索引
		max_result.push_back(number[dq.front()]);
		//保证队列中的第一个值为最大值索引
		for (int i = size; i < number.size(); i++) {
			//队列中的第一值小于滑动窗口的左边界，进行删除
			if(!dq.empty() && dq.front() < i - size + 1) {
				dq.pop_front();
			}
			//多次比较，把每次插入的数与已有的数进行判断，将小于他的数删除
			while (!dq.empty() && number[dq.back()] < number[i]) {
				dq.pop_back();
			}
			dq.push_back(i);
			max_result.push_back(number[dq.front()]);
		}
	}
	return max_result;
}
int main()
{
	vector<int> number = { 10,5,6,8,3,5,9,1,5 };
	int size = 3;
	vector<int> result = sliding_window(number, size);
	for (int i = 0; i < result.size(); i++) {
		cout << result[i] << endl;
	}
	system("pause");
	return 0;
}

运行结果