题目:
一个长度为 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;
}