小y的序列---ST表+二分

题目链接

单调性：
随着区间增大，区间中最大值-区间最小值单调递增

#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';
}