poj_2823 单调队列

题目大意

    给定一行数,共N个。有一个长度为K的窗口从左向右滑动,窗口中始终有K个数字,窗口每次滑动一个数字。求各个时刻窗口中的最大值和最小值。

题目分析

    直接搜索,复杂度为O(n^2)。考虑使用单调队列,单调队列中的元素(或者元素的相关信息)单调递增或者递减。在本题中用一个单调递增的队列A保存当前窗口中值逐渐递增的索引,队列的头部元素为当前窗口中的最小值的索引;用一个单调递减的队列B保存当前窗口中值逐渐递减的索引,队列的头部元素为当前窗口中最大值的索引。 
    窗口每移动到一个新的元素ele时,若新元素大于A中的队尾元素,则入队列,否则,不断弹出队尾的元素,直到队尾元素小于ele,然后入队;若新元素小于B的队尾元素,则入队列,否则,不断弹出队尾元素,直到队尾元素大于ele,然后入队。 
    若窗口移动到的最新元素的index - 队列头元素(队列中存放的是索引) > 窗口长度k,则队列头部后移,以保证队列中的索引位于窗口之内。 
    每个元素最多入栈2次,出栈2次,平摊复杂度为O(n).

    单调队列/栈中存放的一般为序列元素的索引(可能还有其他更多信息),且每次新元素和队尾/栈顶元素比较,若满足单调性质,则入队/入栈;否则,不断弹出队尾/栈顶元素,直到满足单调性质,再入队/入栈。 
    有时可能还需要考虑栈和队列的size(如本题中的窗口)

实现(c++)

#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>
#define MAX_SIZE 1000005
int gMaxInWindow[MAX_SIZE];
int gMinInWindow[MAX_SIZE];
int gIncQ[MAX_SIZE];
int gDecQ[MAX_SIZE];
int gArray[MAX_SIZE];
int main(){
	int n, k;
	scanf("%d %d", &n, &k);
	int inc_front = 0, inc_tail = -1, dec_front = 0, dec_tail = -1;
	int ele;
	for (int i = 0; i < n; i++){
		scanf("%d", &ele);
		gArray[i] = ele;
		
		while (inc_tail >= inc_front && gArray[gIncQ[inc_tail]] >= ele){
			inc_tail--;
		}
		inc_tail++;
		gIncQ[inc_tail] = i;

		while (dec_tail >= dec_front && gArray[gDecQ[dec_tail]] <= ele){
			dec_tail--;
		}
		dec_tail++;
		gDecQ[dec_tail] = i;

		if (i - gIncQ[inc_front] >= k){
			inc_front++;
		}
		if (i - gDecQ[dec_front] >= k){
			dec_front++;
		}
		if (i >= k - 1){
			gMinInWindow[i - k + 1] = gArray[gIncQ[inc_front]];
			gMaxInWindow[i - k + 1] = gArray[gDecQ[dec_front]];
		}
	}
	for (int i = 0; i <= n - k; i++){
		printf("%d ", gMinInWindow[i]);
	}
	printf("\n");
	for (int i = 0; i <= n - k; i++){
		printf("%d ", gMaxInWindow[i]);
	}
	

	return 0;
}

 

posted @ 2015-09-24 16:40  农民伯伯-Coding  阅读(649)  评论(0编辑  收藏  举报