單調隊列：：poj2823 Sliding Window

最新推荐文章于 2024-09-05 23:31:07 发布

原创最新推荐文章于 2024-09-05 23:31:07 发布 · 734 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

算法

本文介绍了一种使用单调队列解决数组窗口滑动问题的方法，包括算法原理、实现细节以及效率对比。通过实例展示了如何利用单调队列优化窗口内最大值和最小值的查找过程。

题目：

一个长度为 N 的数组，一个长为 K 的滑动窗体从最左移至最右端，你只能见到窗口的K个数，每次窗体向右移动一位，如下表：要找出窗口在各位置时的 max value,min value

Window position                 Min value  Max value 
[ 1  3  -1 ]  -3  5  3  6  7    -1         3 
1  [ 3  -1  -3 ]  5  3  6  7    -3         3 
1  3  [ -1  -3  5 ]  3  6  7    -3         5 
1  3  -1  [ -3  5  3 ]  6  7    -3         5 
1  3  -1  -3  [ 5  3  6 ]  7    3          6 
1  3  -1  -3  5  [ 3  6  7 ]    3          7

一开始我想用和3250一样的DP来做，但发现那是行不通的，没有最优子结构。尝试用单调队列：维护一个单调队列，若队列为空，则入队；首先找到队头；然后若新元素比队尾元素小则入队，若比队尾元素大，则队尾不断出队，然后新元素入队；判断此时队长是否大于等于k，若是则输出队头。

自己寫的單調隊列代码在VS上试了很多次都是可以的，但在poj上老是AC不了，没办法，先忍着。

下面是用STL的模板 priority_queue 來做的，簡潔明瞭，耗時差不多，但由於vector占用的內存較大，所以18836k有點蔚為壯觀。

#include "stdafx.h"
#include<iostream>    //18836K 5844MS 
#include<queue>
int MIN(int a,int b) { return a<b?a:b; }
int MAX(int a,int b) { return a>b?a:b; }
#define CLR(NAME,VALUE) memset(NAME,VALUE,sizeof(NAME))
using namespace std;
const int N=1000011;
int d[N],imin[N],imax[N];

struct Cmp1{bool operator()(const int a1,const int a2){return d[a1]>d[a2];}};
struct Cmp2{bool operator()(const int a1,const int a2){return d[a1]<d[a2];}};
priority_queue<int,vector<int>,Cmp1> q1;  //最小堆
priority_queue<int,vector<int>,Cmp2> q2;  //最大堆

int _tmain(int argc, _TCHAR* argv[])
{
	int i,n,k;
	scanf("%d%d",&n,&k);
	if( k>n )  k=n;
	for(i=0;i<n;++i) scanf("%d",d+i);   //輸入整個數組
	for(i=0;i<k;++i){q1.push(i);q2.push(i);}  //初始化堆
	int len(0);       //長度初始化
	imin[len]=d[q1.top()];   //前k個數的最小值
	imax[len++]=d[q2.top()]; //前k個數的最大值
	for(i=k;i<n;++i){
		q1.push(i);
		q2.push(i);
		while( i-q1.top()>=k )  q1.pop();
		while( i-q2.top()>=k )  q2.pop();
		imin[len]=d[q1.top()];
		imax[len++]=d[q2.top()];
	}
	for(i=0;i<len;++i)
		printf("%d%c",imin[i],i==len-1?'\n':' ');
	for(i=0;i<len;++i)
		printf("%d%c",imax[i],i==len-1?'\n':' ');
    
     return 0;
}

标签

#struct #vector #c