题目链接
单调性:
随着区间增大,区间中最大值-区间最小值单调递增
#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
#include <bitset>
#include <map>
using namespace std;
typedef long long LL;
typedef pair<int,int>PII;
const int N = 1000010;
int a[N];
int n,k;
int Fmax[N][20];
int Fmin[N][20];
int lg2[N];
void init_log()
{
lg2[0]=-1;
for(int i=1;i<N;i++)lg2[i]=lg2[i>>1]+1;
}
void init()
{
for(int i=1;i<N;i++)Fmax[i][0]=Fmin[i][0]=a[i];
int k=lg2[n];
for(int j=1;j<=k;j++)
for(int i=1;i<=n-(1<<j)+1;i++)
{
Fmax[i][j]=max(Fmax[i][j-1],Fmax[i+(1<<j-1)][j-1]);
Fmin[i][j]=min(Fmin[i][j-1],Fmin[i+(1<<j-1)][j-1]);
}
}
int RMQ(int l,int r)
{
int k=lg2[r-l+1];
int maxv=max(Fmax[l][k],Fmax[r-(1<<k)+1][k]);
int minv=min(Fmin[l][k],Fmin[r-(1<<k)+1][k]);
return maxv-minv;
}
signed main()
{
cin>>n>>k;
for(int i=1;i<=n;i++)scanf("%d",a+i);
init_log();
init();
LL res=0;
for(int i=1;i<=n;i++)
{
int l=i,r=n;
while(l<r)
{
int mid=l+r>>1;
if(RMQ(i,mid)<k)l=mid+1;
else r=mid;
}
if(RMQ(i,r)!=k)continue;
int pos=r;
l=pos,r=n;
while(l<r)
{
int mid=l+r+1>>1;
if(RMQ(i,mid)>k)r=mid-1;
else l=mid;
}
res+=r-pos+1;
}
cout<<res<<'\n';
}