CF689D Friends and Subsequences(ST表+二分)

题目传送门

题意

给定序列 $a$ 和序列 $b$，长度均为 $n$。问有多少组 $(l,r)$，满足 $1\le l\le r\le n$ 且
$$\underset{i=l}{\overset{r}{\max }}{a}_i=\underset{i=l}{\overset{r}{\min }}{b}_i$$

• $1\le n\le 2\times 10^5$，$|a_i|,|b_i|\le 10^9$。

思路

这个题还是利用了单调性质。$max[a_i]$是一个单调递增的序列，$min[b_i]$是一个单调递减的序列。
因此我们可以进行一个二分来找到max,和min的一个区间，不过这个题需要两头夹，因此要用两次二分来枚举这个区间。

#include<iostream>
#include<cmath>
#include<map>
using namespace std;

typedef long long ll;
int sta[205050][30];
int stb[205050][30];
ll ans=0;

int querya(int l,int r){
	int k=log2(r-l+1);
	//cout<<k<<endl;
	//cout<<l<<" "<<k<<"       "<<l-(1<<k)+1<<" "<<k<<endl; 
	return max(sta[l][k],sta[r-(1<<k)+1][k]);
}
int queryb(int l,int r){
	int k=log2(r-l+1);
	return min(stb[l][k],stb[r-(1<<k)+1][k]);
}
int bsearch_l(int x,int n){
	int l=x,r=n;
	while(l<r){
		int mid=l+r+1>>1;
		if(querya(x,mid)<=queryb(x,mid))l=mid;
		else r=mid-1;
	}
	if(l<=n&&querya(x,l)==queryb(x,l))return l;
	else return -1;
}
int bsearch_r(int x,int n){
	int l=x,r=n;
	while(l<r){
		int mid=l+r>>1;
		if(querya(x,mid)>=queryb(x,mid))r=mid;
		else l=mid+1;
	}
	if(l>=1&&querya(x,l)==queryb(x,l))return l;
	else return -1;
}
int main(){
	int n;
	cin>>n;
	for(int i=1;i<=n;i++)cin>>sta[i][0]; 
	for(int i=1;i<=n;i++)cin>>stb[i][0];
	for(int i=1;(1<<i)<=n;i++){
		for(int j=1;j+(1<<(i-1))<=n;j++){
			sta[j][i]=max(sta[j][i-1],sta[j+(1<<(i-1))][i-1]);
			stb[j][i]=min(stb[j][i-1],stb[j+(1<<(i-1))][i-1]);
		}
	}
	//cout<<querya(4,5)<<" "<<queryb(4,5)<<endl;
	for(int i=1;i<=n;i++){
		int l=bsearch_l(i,n);
		int r=bsearch_r(i,n);
		//cout<<l<<" "<<r<<endl;
		if(l!=-1&&r!=-1){
			swap(l,r);
		//	cout<<l<<" "<<r<<endl;
			ans+=(ll)(r-l+1);
		}
	}
	cout<<ans<<endl;
}