题意比较好理解。刚开始没注意n的范围,,用了暴力,结果超时了。
一个比较好的做法就是使用rmq+二分法来查找某个区间的最值。
dpmax[i][j]来代表从下标i开始,一共2^j个元素中的最大值。
dpmin[i][j]来代表从下标i开始,一共2^j个元素中的最小值。
对于每个元素的下标i,它符合条件的区间为[1,n-i],我们二分这个区间,看看这个区间内的最小值和最大值之间的差值是否符合题目的条件,如果区间的长度为len,那么最终的答案ans += len,一直遍历完所有的元素,复杂度应该为nlog2n,题目限时2000ms,我的程序跑了1900+ms,,真是太悬了。
#include<stdio.h>
#include<iostream>
#include<string>
#include<string.h>
#include<algorithm>
#include<vector>
#include<time.h>
#include<queue>
#include<stack>
#include<iterator>
#include<math.h>
#include<stdlib.h>
#include<limits.h>
#include<map>
//#define ONLINE_JUDGE
#define eps 1e-8
#define INF 0x7fffffff
#define inf 0x3f3f3f3f
#define FOR(i,a) for((i)=0;i<(a);(i)++)
#define MEM(a) (memset((a),0,sizeof(a)))
#define sfs(a) scanf("%s",a)
#define sf(a) scanf("%d",&a)
#define sfI(a) scanf("%I64d",&a)
#define pf(a) printf("%d\n",a)
#define pfI(a) printf("%I64d\n",a)
#define pfs(a) printf("%s\n",a)
#define sfd(a,b) scanf("%d%d",&a,&b)
#define sft(a,b,c)scanf("%d%d%d",&a,&b,&c)
#define for1(i,a,b) for(int i=(a);i<b;i++)
#define for2(i,a,b) for(int i=(a);i<=b;i++)
#define for3(i,a,b)for(int i=(b);i>=a;i--)
#define MEM1(a) memset(a,0,sizeof(a))
#define MEM2(a) memset(a,-1,sizeof(a))
#define ll __int64
const double PI=acos(-1.0);
template<class T> T gcd(T a,T b){return b?gcd(b,a%b):a;}
template<class T> T lcm(T a,T b){return a/gcd(a,b)*b;}
template<class T> inline T Min(T a,T b){return a<b?a:b;}
template<class T> inline T Max(T a,T b){return a>b?a:b;}
using namespace std;
int n,m;
#define N 100010
#define Mod 1000000007
#define maxn 1010
int dpmax[N][20];
int dpmin[N][20];
void RMQ(){
int k = (int)(log(N)*1.0/log(2));
for(int j=1;j<=k;j++){
for(int i=1;i+(1<<j)-1<=n;i++){ //dp[i][j]代表从第i个元素开始的一共2^j个元素中的最值
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 maxQuery(int i,int len){
int j = i+len-1; //右区间的值
int k =(int)(log(j-i+1)*1.0/log(2)); //计算i和j之间一共有几个值,它是2的几次方
return Max(dpmax[i][k],dpmax[j-(1<<k)+1][k]);
}
int minQuery(int i,int len){
int j = i+len-1;
int k =(int)(log(j-i+1)*1.0/log(2.0));
return Min(dpmin[i][k],dpmin[j-(1<<k)+1][k]);
}
int main(){
#ifndef ONLINE_JUDGE
freopen("in.txt","r",stdin);
// freopen("out.txt","w",stdout);
#endif
int t;
sf(t);
while(t--){
sfd(n,m);
int x;
for(int i=1;i<=n;i++){
sf(x);
dpmax[i][0] = x;
dpmin[i][0] = x;
}
RMQ();
ll ans = 0;
for(int i=1;i<=n;i++){
int l = 1,r = n-i+1; //l是开始元素的下标,r是复合条件区间的长度
while(l<=r){
int mid = (l+r)/2;
int mmax = maxQuery(i,mid);
int mmin = minQuery(i,mid);
int dif = mmax-mmin;
if(dif<m)
l = mid+1;
else
r = mid-1;
}
ans += (ll)(l-1);
}
printf("%I64d\n",ans);
}
return 0;
}