hdu 5289 Assignment(15多校第一场1002)

//15多校第一场1002
//RMQ+二分
//求最大值最小值小于k的区间数。
//RMQ预处理区间的极值。
//然后通过枚举区间左端点，二分右端点进行累加。
//每次累加的个数为区间长度（R-L+1）；
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long LL;
#define MIN(a,b) a<b?a:b
#define MAX(a,b) a>b?a:b
const int N=100010,INF=0x3fffffff;
LL a[N];
LL dpmin[N][30],dpmax[N][30];

void RMQ_init(int n){
    for(int i=0;i<n+1;i++)
        dpmax[i][0]=dpmin[i][0]=a[i];
    for(int j=1;(1<<j)<=n+1;j++){
        for(int i=0;i+(1<<j)-1<=n;i++){
            dpmax[i][j]=MAX(dpmax[i][j-1],dpmax[i+(1<<(j-1))][j-1]);
            dpmin[i][j]=MIN(dpmin[i][j-1],dpmin[i+(1<<(j-1))][j-1]);
        }
    }
}
int RMQ_max(int L,int R){
    int k=0;
    while(1<<(k+1)<=R-L+1)k++;
    return MAX(dpmax[L][k],dpmax[R-(1<<k)+1][k]);
}
int RMQ_min(int L,int R){
    int k=0;
    while(1<<(k+1)<=R-L+1)k++;
    return MIN(dpmin[L][k],dpmin[R-(1<<k)+1][k]);
}
int getR(int L,int n,int k){
    int l=L,r=n;
    int ans=0;
    while(l<=r){
        int mid=(l+r)>>1;
        int nn=RMQ_max(L,mid)-RMQ_min(L,mid);
        if(nn<k){
            l=mid+1;
            ans=mid;
        }
        else{
            r=mid-1;
        }
    }
    return ans;
}
int main(){
    int T;
    scanf("%d",&T);
    int n,k;
    while(T--){
        scanf("%d%d",&n,&k);
        for(int i=1;i<=n;i++){
            scanf("%lld",&a[i]);
        }
        RMQ_init(n);
        long long tot=0;
        for(int L=1;L<=n;L++){
            int R=getR(L,n,k);
            tot+=(R-L+1);
        }
        printf("%lld\n",tot);
    }
    return 0;
}