【POI2010】【BZOJ2096】Pilots

Description

Tz又耍畸形了!!他要当飞行员,他拿到了一个飞行员测试难度序列,他设定了一个难度差的最大值,在序列中他想找到一个最长的子串,任意两个难度差不会超过他设定的最大值。耍畸形一个人是不行的,于是他找到了你。
Input

输入:第一行两个有空格隔开的整数k(0<=k<=2000,000,000),n(1<=n<=3000,000),k代表Tz设定的最大值,n代表难度序列的长度。第二行为n个由空格隔开的整数ai(1<=ai<=2000,000,000),表示难度序列。
Output

输出:最大的字串长度。
Sample Input

3 9

5 1 3 5 8 6 6 9 10
Sample Output

4

(有两个子串的长度为4: 5, 8, 6, 6 和8, 6, 6, 9.最长子串的长度就是4)

HINT

Source

by poi

考虑维护两个单调队列,队列里元素是下标关键字是对应下标元素大小
一个递增一个递减
枚举右端点如果当前最大值与最小值的差大于k,调整一下左端点

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXN 3000010
#define GET (ch>='0'&&ch<='9')
using namespace std;
int n,k,ans;
int a[MAXN];
int q1[MAXN],q2[MAXN];
int h1,t1,h2,t2;
void in(int &x)
{
    char ch=getchar();x=0;
    while (!GET)    ch=getchar();
    while (GET) x=x*10+ch-'0',ch=getchar();
}
int main()
{
    in(k);in(n);int tmp=1;
    for (int i=1;i<=n;i++)
    {
        in(a[i]);
        while (h1!=t1&&a[i]>=a[q1[t1-1]])   t1--;
        while (h2!=t2&&a[i]<=a[q2[t2-1]])   t2--;
        q1[t1++]=i;q2[t2++]=i;
        while (a[q1[h1]]-a[q2[h2]]>k)
            if (q1[h1]<q2[h2])  tmp=q1[h1]+1,h1++;
            else    tmp=q2[h2]+1,h2++;
        ans=max(ans,i-tmp+1);
    }
    printf("%d\n",ans);
}

你可能感兴趣的:(单调队列)