解题思路:这里使用双端队列来维护最大值和最小值,然后结合尺取法来解题。其中low维护的是到1--i个数中的最小值然后,high同理。然后不断增加i, 同时j不变,直到最大-最小>=k为止,此时[j,i-1]还是可行的,那么以j为开头,末尾任选有i-j种。
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
deque<int> low,high;
int s[100009];
int main()
{
//freopen("t.txt","r",stdin);
int n,k,T,j;
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&n,&k);
while(!high.empty()) high.pop_back();
while(!low.empty()) low.pop_back();
for(int i=0;i<n;i++)
{
scanf("%d",&s[i]);
}
int d=0;
long long res=0;
j=0;
for(int i=0;i<n;i++)
{
while(!high.empty()&&s[i]>high.back()) high.pop_back();
high.push_back(s[i]);
while(!low.empty()&&s[i]<low.back()) low.pop_back();
low.push_back(s[i]);
while(!high.empty()&&!low.empty()&&high.front()-low.front()>=k)
{
res+=(i-j);
if(high.front()==s[j]) high.pop_front();
if(low.front()==s[j]) low.pop_front();
j++;
}
}
while(j<n)
{
res+=(n-j);
j++;
}
printf("%lld\n",res);
}
return 0;
}