BZOJ2096:[POI2010]Pilots

浅谈队列:https://www.cnblogs.com/AKMer/p/10314965.html

题目传送门:https://lydsy.com/JudgeOnline/problem.php?id=2096

尺取法,详见这篇博客:https://www.cnblogs.com/AKMer/p/10323600.html

不过这次我们要统计的区间越长越好。

所以我们需要在最大值减最小值不超过给定的\(k\)的情况下尽量让左端点靠左。

用一个单调队列维护区间最大值,另一个单调队列维护区间最小值。

如果最大值减最小值大于\(k\)那么把位置靠前的那个值弹出队列,那个位置的后边一个就是新的左端点。

时间复杂度:\(O(n)\)

空间复杂度:\(O(n)\)

代码如下:

#include <cstdio>
#include <algorithm>
using namespace std;

const int maxn=3e6+5;

int a[maxn],list1[maxn],list2[maxn];
int n,k,ans,pos,head1,tail1,head2,tail2;

int read() {
    int x=0,f=1;char ch=getchar();
    for(;ch<'0'||ch>'9';ch=getchar())if(ch=='-')f=-1;
    for(;ch>='0'&&ch<='9';ch=getchar())x=x*10+ch-'0';
    return x*f;
}

int main() {
    k=read(),n=read();
    for(int i=1;i<=n;i++)a[i]=read();
    for(int i=1;i<=n;i++) {
        while(head1!=tail1&&a[list1[tail1-1]]<=a[i])tail1--;
        list1[tail1++]=i;
        while(head2!=tail2&&a[list2[tail2-1]]>=a[i])tail2--;
        list2[tail2++]=i;
        while(a[list1[head1]]-a[list2[head2]]>k) {
            if(list1[head1]<list2[head2])pos=list1[head1++];
            else pos=list2[head2++];
        }
        ans=max(ans,i-pos);
    }
    printf("%d\n",ans);
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/AKMer/p/10323769.html