【题目描述】
【思路】
固定左端点
i
i
i,向右寻找一个最远的右端点
j
j
j ,使得区间
a
[
i
,
j
]
a[i,j]
a[i,j] 中的最大值减去最小值的差
<
=
k
<=k
<=k 同时
a
[
i
,
j
+
1
]
a[i,j+1]
a[i,j+1] 中的最大值减去最小值的差
>
k
>k
>k,这样一来
[
i
,
i
]
,
[
i
,
i
+
1
]
.
.
.
[
i
,
j
]
[i,i],[i,i+1]...[i,j]
[i,i],[i,i+1]...[i,j] 都符合要求,以
i
i
i 为左端点的区间对答案的贡献为
j
−
i
+
1
j-i+1
j−i+1,因为随着
i
i
i 的增加,
j
j
j 也只能不断增加,所以可以用单调队列求区间的最大值和最小值
#include<bits/stdc++.h>
using namespace std;
const int maxn=50005;
int n,k;
int a[maxn];
deque<int> minq,maxq;
int main(){
scanf("%d%d",&n,&k);
for(int i=0;i<n;++i) scanf("%d",&a[i]);
long long ans=0;
int ri=0;
for(int i=0;i<n;++i){
while(minq.size() && minq.front()<i) minq.pop_front();
while(maxq.size() && maxq.front()<i) maxq.pop_front();
while(ri<n){
while(minq.size() && a[minq.back()]>=a[ri]) minq.pop_back();
minq.push_back(ri);
while(maxq.size() && a[maxq.back()]<=a[ri]) maxq.pop_back();
maxq.push_back(ri);
if(a[maxq.front()]-a[minq.front()]>k) break;
++ri;
}
ans+=ri-i;
}
printf("%lld\n",ans);
return 0;
}