题意:
给你一个序列,让你求出符合(区间最大值-区间最小值)<=k的的区间有多少个。
序列大小200000
理解:
很显然,n^2 会挂,
那么用st表,或是线段树,什么的维护个区间最值,就行了,
查询所有区间肯定不能一个一个查,因为区间数也是n^2.因为区间最值差是单调非减的,所以二分以下区间最值差就可以了。
O(nlogn)
#include <bits/stdc++.h>
using namespace std;
const int maxn = 2e5+5;
const int mod=1e9+7;
#define ll long long
inline int read(){int x=0,f=1;char c=getchar();for(;!isdigit(c);c=getchar())
if(c=='-')f=-1;for(;isdigit(c);c=getchar())x=x*10+c-'0';return x*f;}
int dp[maxn][20];
int dd[maxn][20];
int mm[maxn];
void init(int n,int b[]){
mm[0]=-1;
for(int i=1;i<=n;++i){
mm[i]=((i&(i-1))==0)?mm[i-1]+1:mm[i-1];
dp[i][0]=b[i];
dd[i][0]=b[i];
}
for(int j=1;j<=mm[n];++j)
for(int i=1;i+(1<<j)-1<=n;++i){
dp[i][j]=max(dp[i][j-1],dp[i+(1<<(j-1))][j-1]);
dd[i][j]=min(dd[i][j-1],dd[i+(1<<(j-1))][j-1]);
}
}
int rmq1(int x,int y){
int k=mm[y-x+1];
return max(dp[x][k],dp[y-(1<<k)+1][k]);
}
int rmq2(int x,int y){
int k=mm[y-x+1];
return min(dd[x][k],dd[y-(1<<k)+1][k]);
}
int a[maxn];
int n,k;
bool judge(int L,int R){
int mx=rmq1(L,R);
int mn=rmq2(L,R);
if(mx-mn<=k) return true;
return false;
}
int main(){
n=read();k=read();
for(int i=1;i<=n;++i) a[i]=read();
init(n,a);
ll ans=0;
int t1,t2;
for(int i=1;i<=n;++i){
int l=i,r=n,m;
while(l<=r){
m=(l+r)/2;
if(judge(i,m)) l=m+1;
else r=m-1;
}
ans+=l-i;
}
printf("%lld\n",ans);
return 0;
}