题意:给一个数字序列然后问你有多少个区间的最大值与最小值的差小于k,输出
思路:对于最大值与最小值的差我们直接用RMQ的ST可以快速求得,然后求当前的i到后面哪里是都可以的,这个与尺取的思想如出一辙,直接求出来就行了,输出注意用long long 就行了,没什么难的
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int maxn=100010;
int max1[maxn][20],min1[maxn][20],num[maxn],ans[maxn];
int n,m;
void RMQ_init(){
for(int i=1;i<=n;i++)
max1[i][0]=min1[i][0]=num[i];
for(int i=1;(1<<i)<=n;i++){
for(int j=1;j+(1<<i)-1<=n;j++){
max1[j][i]=max(max1[j][i-1],max1[j+(1<<(i-1))][i-1]);
min1[j][i]=min(min1[j][i-1],min1[j+(1<<(i-1))][i-1]);
}
}
}
int RMQ(int le,int ri){
int k=0;
while((1<<(k+1))<=ri-le+1) k++;
int ans1=max(max1[le][k],max1[ri-(1<<k)+1][k]);
int ans2=min(min1[le][k],min1[ri-(1<<k)+1][k]);
return ans1-ans2;
}
void slove(){
int s=1,t=1;
while(1){
while(t<=n&&RMQ(s,t)<m) t++;
if(t>n) break;
ans[s]=t-s;
s++;
}
for(;s<=n;s++) ans[s]=n-s+1;
}
int main(){
int T;
scanf("%d",&T);
while(T--){
scanf("%d%d",&n,&m);
memset(ans,0,sizeof(ans));
for(int i=1;i<=n;i++) scanf("%d",&num[i]);
RMQ_init();
slove();
ll sum=0;
for(int i=1;i<=n;i++) sum+=(ll)ans[i];
printf("%I64d\n",sum);
}
return 0;
}