POJ 2823 Sliding Window(单调队列)

/*
单调队列
适合解决的问题,多次查询k个连续序列中最大或最小值。可以将复杂度从O(n*n)缩短到O(n)。
实现模式:队列实现,只不过其中元素单调(依次增大或减小),我们假设求最大值。入队时比较队尾元素与插入元素的大小,如果队尾元素小与插入元素,则对尾元素出对,直到大于等于位置。
这样插入k个元素后,队列中队首即为最大值。继续进行第二次查询时,需要比较队首元素在原来元素中的位置,判断是否在这次查询的区间内,如果不在,则出对。使用自定义数组即可实现。
*/

#include <iostream>
using namespace std;
const int nMax = 1000100;
struct Queue
{
	int pos;
	int w;
	Queue(){}
	Queue(int pos, int w): pos(pos), w(w){}
}queue[nMax];
int N, K;
int A[nMax];

void solve1()
{
	int front, rear;
	front = rear = 0;
	int i;
	for(i = 1; i <= N; ++ i)
	{
		while(front != rear && A[i] < queue[front - 1].w) front --;
		queue[front ++] = Queue(i, A[i]);
		if(i >= K)
		{
			while(queue[rear].pos <= i - K)
				rear++;
			if(i != K)
				printf(" ");
			printf("%d", queue[rear].w);
		}
	}
	printf("\n");
}

void solve2()
{
	int front, rear;
	front = rear = 0;
	int i;
	for(i = 1; i <= N; ++ i)
	{
		while(front != rear && A[i] > queue[front - 1].w) front --;
		queue[front ++] = Queue(i, A[i]);
		if(i >= K)
		{
			while(queue[rear].pos <= i - K)
				rear++;
			if(i != K)
				printf(" ");
			printf("%d", queue[rear].w);
		}
	}
	printf("\n");
}

int main()
{
	while(scanf("%d%d", &N, &K) != EOF)
	{
		int i;
		for(i = 1; i <= N; ++ i) scanf("%d", &A[i]);
		solve1();
		solve2();
	}
	return 0;
}


你可能感兴趣的:(POJ 2823 Sliding Window(单调队列))