题意:给一个数组,给一个K,找到满足连续子串的个数,使子串中的最小值与最大值之差小于K, N:1e5
思路:确定右端点,二分枚举右端点,找到最大的满足位置,用RMQ判断,统计每一左端点的贡献即可;
AC代码:
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long LL;
const int MAXN=1e5+11;
int a[MAXN],dmin[MAXN][33],dmax[MAXN][33],prelog2[MAXN+11];
void RMQ_init(int N) {
for(int i=0;i<N;++i) dmin[i][0]=dmax[i][0]=a[i];
for(int j=1;(1<<j)<=N;++j) for(int i=0;i+(1<<j)-1<N;++i) {
dmin[i][j]=min(dmin[i][j-1],dmin[i+(1<<j-1)][j-1]);
dmax[i][j]=max(dmax[i][j-1],dmax[i+(1<<j-1)][j-1]);
}
}
int RMQ(int L,int R,int cho) {
int k=prelog2[R-L+1];//log(1.0*(R-L+1))/log(2.0); log超时
//while((1<<k+1)<=R-L+1) ++k;
if(cho ) return max(dmax[L][k],dmax[R-(1<<k)+1][k]);
return min(dmin[L][k],dmin[R-(1<<k)+1][k]);
}
bool Judge(int L,int mid,int K) {
int diff=RMQ(L,mid,1)-RMQ(L,mid,0);
if(diff<K) return true;
return false;
}
int main() {
prelog2[1]=0;
for(int i=2;i<MAXN-2;++i) {
prelog2[i]=prelog2[i-1];
if((1<<prelog2[i]+1)==i) ++prelog2[i];
}
int T; scanf("%d",&T);
while(T--) {
int N,K; scanf("%d%d",&N,&K);
for(int i=0;i<N;++i) scanf("%d",&a[i]);
RMQ_init(N);//多次RMQ_init() 打log2的表优化还是比较明显的
LL ans=0;
int i=0;
while(i<N) {
int L=i,R=N,pos=N;
while(L<R) {
int mid=L+R>>1;
if(Judge(i,mid,K)) {
// if(pos==mid) break;
pos=mid;
L=mid+1;
}
else R=mid;
}
// printf("pppp %d\n",pos);
LL len=pos-i+1;
ans=ans+len;
++i;
}
printf("%lld\n",ans);
}
return 0;
}