解题思路
构建两个单调队列 m 1 , m 2 m1,m2 m1,m2,分别维护区间 [ i , j ] [i,j] [i,j]里的最大值和最小值,枚举i作为区间的头, j j j作为一个指针一直向后移动,同时把数分别插入队尾,当 m 1 [ h ] − m 2 [ h ] > k m1[h]-m2[h]>k m1[h]−m2[h]>k时退出,然后判断队头是否当前区间的头,如果是,就弹出。
代码
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#define ll long long
using namespace std;
int n,k,ans,a[50010];
int main() {
scanf("%d%d",&n,&k);
for(int i=0; i<n; i++)
scanf("%d",&a[i]);
deque<int>st1,st2;
for(int i=0,j=0; i<n; i++) {
while(j<n)
{
while(!st1.empty()&&a[j]>=a[st1.back()])
st1.pop_back();
while(!st2.empty()&&a[j]<=a[st2.back()])
st2.pop_back();
st1.push_back(j);
st2.push_back(j);
if(a[st1.front()]-a[st2.front()]<=k)
j++;
else break;
}
ans+=(j-i);
if(st1.front()==i)
st1.pop_front();
if(st2.front()==i)
st2.pop_front();
}
printf("%d",ans);
}