【复赛模拟试题】河床

【问题描述】

　　地理学家们经常要对一段河流进行测量分析。他们从上游开始向下游方向等距离地选择n(n<=30000)个点测量水位深度。得到一组数据a1,a2,…,an，回到实验室后数据分析员根据需要对数据进行分析，发掘隐藏在数据背后的规律。

　　最近，乌龙博士发现某种水文现象与河床地势有关，于是他指示分析员要找出一段河流中最大高低起伏差不超过K(1<=K<=100)的最长的一段。这看似一个简单的问题，由于任务紧急，分析员来求助于你，并告诉你博士的所有数据都精确到个位。

【输入格式】

　　第一行包含两个整数N和K，分别表示测量点的个数和博士要求的最大水深度（也就是河床的地势差）。
　　第二行包含N个整数，表示从上游开始依次得到的水位深度di(1<=i<=n,0<=di<=32767)

【输出格式】

　　一个整数m，表示最长一段起伏不超过K的河流长度，用测量点的个数表示

【输入样例】

6 2
5 3 2 2 4 5

【输出样例】

4

【数据范围】

30%的数据，满足：1<=n<=5000；
50%的数据，满足：1<=n<=30000，符合条件的最长连续序列长度不超过100；
100%的数据，满足：1<=n<=30000，符合条件的最长连续序列长度不超过n；

【来源】

从第2个测量点到第5个测量点之间的一段，即3 2 2 4，起伏最大为2

这道题我用的滑动窗口（用优先队列优化），动态维护最大值和最小值。这样就可以很快的枚举出ans了。
代码如下：

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<queue>
#include<cstring>
#include<vector>
using namespace std;
const int maxn=30005;
const int inf=100000000;
int n,k,a[maxn],d[3][maxn]={0},ans=0;
struct shu
{
    int h,id;
    friend bool operator <(shu a,shu b)
    {
        return a.h<b.h;
    }
};
struct shu1
{
    int h,id;
    friend bool operator <(shu1 a,shu1 b)
    {
        return a.h>b.h;
    }
};
priority_queue<shu>q;
priority_queue<shu1>q1;
int read()
{
    int x=0;
    char ch=getchar();
    while(ch<'0'||ch>'9') ch=getchar();
    while(ch<='9'&&ch>='0')
    {
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x;
}
void init()
{
    n=read();k=read();
    for(int i=1;i<=n;i++)
    a[i]=read(); 
}
int main()
{
    //freopen("river.in","r",stdin);
    //freopen("river.out","w",stdout);
    init();
    int s=1;
    q.push((shu){a[1],1});
    q1.push((shu1){a[1],1});
    ans=1;
    for(int i=2;i<=n;i++)
    {
        q.push((shu){a[i],i});
        q1.push((shu1){a[i],i});
        shu t=q.top();
        shu1 tt=q1.top();
        while(t.h-tt.h>k)
        {
            if(t.id<=tt.id)
            {
                s=max(s,t.id+1);
                q.pop();
            }
            if(t.id>tt.id) 
            {
                s=max(s,tt.id+1);
                q1.pop();
            }
            t=q.top();
            tt=q1.top();
        }
        ans=max(ans,i-s+1);
    }
    printf("%d",ans);
    return 0;
}