滑动窗口(单调队列的应用)

题意:

有一个长度为 n 的数列和一个大小为 k 的窗口,窗口可以在数列上来回移动。求:窗口在从左往右滑的时候,每次窗口内数的最大值和最小值分别是多少。
例如下表中,数列是 [1 3 -1 -3 5 3 6 7],其中 k 等于 3。

Window positionMinimum valueMaximum value
[1 3 -1] -3 5 3 6 7-13
1 [3 -1 -3] 5 3 6 7-33
1 3 [-1 -3 5] 3 6 7-35
1 3 -1 [-3 5 3] 6 7-35
1 3 -1 -3 [5 3 6] 7-36
1 3 -1 -3 5 [3 6 7]-36

输入要求:
输入两行,第一行两个整数 n 和 k 分别表示数列长度和滑动窗口的大小,1<=k<=n<=1000000。第二行输入 n 个整数表示数列。
输出要求:
输出两行,第一行输出滑动窗口在从左到右的每个位置时,滑动窗口中的最小值,第二行输出最大值。
Sample Input:

8 3
1 3 -1 -3 5 3 6 7

Sample Output:

-1 -3 -3 -3 3 3
3 3 5 5 6 7

思路:

  • 首先想到暴力解法,枚举所有窗口的情况,显然,复杂度过高,接近 O( n 2 n^2 n2),所以另寻他法。
  • 这里使用的单调队列,通过单调递增队列和单调递减队列,分别求出每个窗口中的最小值和最大值。
  • 从数组第一位开始,先将 k 位数插入单调队列中,并将队首的值记录,然后从第 k+1 位开始,每次从队尾插入一个元素,再从队首删除一个元素,记录队首,直到整个数列全部插入。
  • 队列用数组 a 和队首 front 和队尾 back 进行记录。单调队列则体现在向队列中插入数据时,先比较队尾元素和插入元素大小再进行删除或插入操作。

代码:

#include <iostream>
using namespace std;

pair<int, int> a[1000005];
pair<int, int> windows[1000005];
int maxAns[1000005];
int minAns[1000005];
int back = -1, front = 0;

void push_back_up(pair<int, int> p) {//单调递增队列
	if (back < front) {
		back++;
		windows[back] = p;
		return;
	}
	while (1) {
		if (back < front) {//若删空了队列
			back++;
			windows[back] = p;
			return;
		}
		if (p.second > windows[back].second) {
			back++;
			windows[back] = p;
			break;
		}
		else {
			back--;
		}
	}
}

void push_back_down(pair<int, int> p) {//单调递减队列
	if (back < front) {
		back++;
		windows[back] = p;
		return;
	}
	while (1) {
		if (back < front) {//若删空了队列
			back++;
			windows[back] = p;
			return;
		}
		if (p.second < windows[back].second) {
			back++;
			windows[back] = p;
			break;
		}
		else {
			back--;
		}
	}
}

int main()
{
	int n, k;
	scanf_s("%d%d", &n, &k);
	for (int i = 1; i <= n; i++) {
		a[i].first = i;
		scanf_s("%d", &a[i].second);
	}

	//求各个区间最小值
	int i = 1;
	for (i; i <= k - 1; i++) {
		push_back_up(a[i]);
	}
	while (i <= n) {
		push_back_up(a[i]);
		if (windows[front].first < i - k + 1) {
			front++;
		}
		minAns[i - k] = windows[front].second;
		i++;
	}
	for (int i = 0; i < n - k + 1; i++) {
		printf("%d ", minAns[i]);
	}
	printf("\n");

	//求各个区间最大值
	back = -1;
	front = 0;
	i = 1;
	for (i; i <= k - 1; i++) {
		push_back_down(a[i]);
	}
	while (i <= n) {
		push_back_down(a[i]);
		if (windows[front].first < i - k + 1) {
			front++;
		}
		maxAns[i - k] = windows[front].second;
		i++;
	}
	for (int i = 0; i < n - k + 1; i++) {
		printf("%d ", maxAns[i]);
	}
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值