首页 > Windows开发 > 详细

POJ 2823 Sliding Window 单调队列

时间:2017-07-21 19:37:16      阅读:238      评论:0      收藏:0      [点我收藏+]

解题思路:

维护一个递增的单调队列和一个递减的单调队列,基础题。

代码:

#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <stack>
#define LL long long
#define FOR(i,x,y) for(int i=x;i<=y;i++)
using namespace std;
const int maxn = 1000000 + 10;
int N , M;
int A[maxn] , Q[maxn] , ID[maxn];
int head , tail;
int len1 , len2;
void init()
{
	FOR(i,1,N) scanf("%d",&A[i]);
	len1 = len2 = 0;
}
void solve_min()
{
	head = 1 , tail = 0;
	for(int i=1;i<M;i++)
	{
		while(head <= tail && Q[tail] >= A[i]) tail--;
		tail++;
		Q[tail] = A[i] ; ID[tail] = i;
	}
	FOR(i,M,N)
	{
		while(head <= tail && Q[tail] >= A[i]) tail--;
		tail++;
		Q[tail] = A[i] ; ID[tail] = i;
		while(ID[head] <= i - M) head++;
		printf("%d ",Q[head]);
	}
}
void solve_max()
{
	head = 1 , tail = 0;
	for(int i=1;i<M;i++)
	{
		while(head <= tail && Q[tail] <= A[i]) tail--;
		tail++;
		Q[tail] = A[i] ; ID[tail] = i;
	}
	FOR(i,M,N)
	{
		while(head <= tail && Q[tail] <= A[i]) tail--;
		tail++;
		Q[tail] = A[i] ; ID[tail] = i;
		while(ID[head] <= i - M) head++;
		printf("%d ",Q[head]);
	}
}
int main()
{
	while(scanf("%d%d",&N,&M)!=EOF)
	{
		init();
		solve_min();printf("\n");
		solve_max();printf("\n");
	}
	return 0;
}

POJ 2823 Sliding Window 单调队列

原文:http://www.cnblogs.com/clnchanpin/p/7219284.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!