牛客练习赛96 C.小y的序列

单调栈+线段树

链接：C-小y的序列_牛客练习赛96 (nowcoder.com)
题意：给定序列 a ，求序列 a 的 $\frac{(n+1)\ast n}{2}$ 个区间中有多少个区间满足区间最大值和区间最小值的差为 k 。

题解：先特殊处理 k 为 0 的情况。然后从左到右枚举所有区间左端点，那么对于一个固定的左端点而言，其左端点越往左，那么区间最大值会越大，区间最小值也会越小。可以用单调栈来表达这一过程。维护两个单调栈，当加入一个新点时，也就是 r 往右移了一位，这时对于最小值单调递增与最大值单调递减的栈都需要做出改变。以最小值单调递增为例，加入新点会使得以该点为右端点的前边区间的最小值进行更新，只更新那些比当前这个加入点值大的位置，因为加入新的右端点后前面某些位置的最小值将改变，每次改变对一个区间进行修改。可以用线段树来维护。线段树维护区间最大值与最小值的差值，然后更新就区间修改，查询的时候就查区间内等于 k 的值有多少个。由于差值从左往右单调递减的，所以查询本质上也是查一个区间，复杂度依然是 $O(nlogn)$ 。

#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
#include<iostream>
#include<algorithm>
#include<vector>
#include<cstring>
#include<functional>
#include<queue>
#include<unordered_map>
#include<stack>
#include<map>
#include<set>

using namespace std;
using ll=long long;
using P=pair<int,int>;
const ll inf=1e18;

struct tree{
	vector<int>mx,mn,tg;
	tree(int x):mx(x+5<<2,0),mn(x+5<<2,0),tg(x+5<<2,0){}
	void pushup(int k){
		mx[k]=max(mx[k<<1],mx[k<<1|1]);
		mn[k]=min(mn[k<<1],mn[k<<1|1]);
	}
	void down(int k,int w){
		mx[k]+=w,mn[k]+=w,tg[k]+=w;
	}
	void pushdown(int k)
	{
		if(tg[k])
		{
			down(k<<1,tg[k]);
			down(k<<1|1,tg[k]);
			tg[k]=0;
		}
	}
	void update(int k,int l,int r,int ql,int qr,int w){
		if(ql<=l&&r<=qr){down(k,w); return;}
		int mid=l+r>>1;
		pushdown(k);
		if(ql<=mid)update(k<<1,l,mid,ql,qr,w);
		if(mid<qr)update(k<<1|1,mid+1,r,ql,qr,w);
		pushup(k);
	}
	int query(int k,int l,int r,int w){
		if(mx[k]==mn[k])return (mx[k]==w)*(r-l+1);
		int mid=l+r>>1,res=0; pushdown(k);
		if(mn[k<<1]<=w&&w<=mx[k<<1])res+=query(k<<1,l,mid,w);
		if(mn[k<<1|1]<=w&&w<=mx[k<<1|1])res+=query(k<<1|1,mid+1,r,w);
		return res;
	}
};

void solve()
{
	int n,k; cin>>n>>k;
	vector<int>a(n+1);
	for(int i=1;i<=n;i++)cin>>a[i];

	if(!k)
	{
		ll cnt=0;
		for(int i=1,j=0;i<=n;i++)
		{
			if(a[i]==a[i-1])j++;
			else 
			{
				cnt+=1ll*j*(j+1)/2;
				j=1;
			}
			if(i==n)
			{
				cnt+=1ll*j*(j+1)/2;
			}
		}
		cout<<cnt<<"\n"; return;
	}

	tree t(n);
	stack<P>mx,mn;
	mx.push(P(0,0)),mn.push(P(1e9,0));

	ll ans=0;

	for(int i=1,x,y;i<=n;i++)
	{
		int now=0,pos=i;
		while(mx.top().first>a[i])
		{
			tie(x,y)=mx.top(); mx.pop();
			t.update(1,1,n,y+1,pos,now-a[i]);
			now=x,pos=y;
		}
		t.update(1,1,n,mx.top().second+1,pos,now-a[i]);
		mx.push(P(a[i],i));
		now=0,pos=i;
		while(mn.top().first<a[i])
		{
			tie(x,y)=mn.top(); mn.pop();
			t.update(1,1,n,y+1,pos,a[i]-now);
			now=x,pos=y;
		}
		t.update(1,1,n,mn.top().second+1,pos,a[i]-now);
		mn.push(P(a[i],i));
		ans+=t.query(1,1,n,k);
	}

	cout<<ans<<"\n";

}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0); cout.tie(0);
	int t=1; //cin>>t;
	while(t--)solve();
	return 0;
}