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)
维护两个单调队列,一个单增,一个单减
#include<iostream>
#include<cstdlib>
#include<cstdio>
using namespace std;
const int N=3000005;
int a[N],n,q1[N],l1,r1,q2[N],l2,r2,k,ans,w1[N],w2[N],wz;
int main (){
scanf ("%d%d",&k,&n);
r1=r2=-1;
for (int i=1;i<=n;++i)scanf ("%d",&a[i]);
for (int i=1;i<=n;++i){
for (;l1<=r1 && q1[r1]>=a[i];--r1);q1[++r1]=a[i];w1[r1]=i;
for (;l2<=r2 && q2[r2]<=a[i];--r2);q2[++r2]=a[i];w2[r2]=i;
for (;q2[l2]-q1[l1]>k;wz=(w1[l1]>w2[l2])?(w2[l2++]):(w1[l1++]));
ans=max(ans,i-wz);
}
printf ("%d",ans);
return 0;
}
网友评论