Codeforces 193D. Two Segments

题意：

       给定一个全排列，求有多少个区间[l,r]，使得可以在全排列中选择两个连续序列，使得两个序列合并并排序后恰好是[l,r]。

题解：

       记f(l,r)表示要在全排列中恰好选到[l,r]区间的数，至少要多少个连续序列。显然，当f(l,r)<=2且l<r时是合法区间。

       考虑从f(l,r)转移到f(l,r+1)。

             若(r+1)不和f(l,r)中已有的区间相邻， 那么f(l,r+1)=f(l,r)+1

             若(r+1)和f(l,r)中已有的一个区间相邻，那么f(l,r+1)=f(l,r)

             若(r+1)连接了f(l,r)中已有的两个区间，那么f(l,r+1)=f(l,r)-1

       那么可以从小到大一次加入每一个数，处理出所有的f(l,r)，就得到O(n^2)的算法。

       然后考虑用数据结构优化加入数并统计答案的过程。

       对右端点开线段树。已经知道当左端点是l时的线段树，要求左端点是l+1时的线段树，只需要将一段区间+1，一段区间-1即可。于是问题转化为：每次对一段区间+1或-1，求权值<=2的点的个数。

       这样每个节点维护最小值，次小值及他们的出现次数即可。

       一个小trick是次小值如果存在，一定是最小值+1。

       最后答案要减去n，是因为会把[l,l]这种区间算入。

#include<bits/stdc++.h>
#define lch (x<<1)
#define rch (x<<1|1)
using namespace std;
const int N=3e5+10;

int n,a[N],pl[N];
int m[N<<2],f[N<<2],g[N<<2],d[N<<2];   //m:最小值  f:最小值出现次数  g:次小值出现次数  d:懒标记 

void Pushup(int x) {
	m[x]=min(m[lch],m[rch]);
	f[x]=f[lch]*(m[x]==m[lch])+f[rch]*(m[x]==m[rch]);
	g[x]=g[lch]*(m[x]==m[lch])+g[rch]*(m[x]==m[rch])+f[lch]*(m[x]==m[lch]-1)+f[rch]*(m[x]==m[rch]-1);
}

void Pushdown(int x) {
	m[lch]+=d[x]; m[rch]+=d[x];
	d[lch]+=d[x]; d[rch]+=d[x];
	d[x]=0;
}

void Build(int x,int l,int r) {
	if (l==r) return (void) (f[x]=1);
	int mid=(l+r)>>1;
	Build(lch,l,mid); Build(rch,mid+1,r);
}

void Updata(int x,int l,int r,int lt,int rt,int del) {
	if (lt<=l&&r<=rt) {
		m[x]+=del; d[x]+=del;
		return;
	}
	if (d[x]) Pushdown(x);
	int mid=(l+r)>>1;
	if (mid>=lt) Updata(lch,l,mid,lt,rt,del); 
	if (mid<rt) Updata(rch,mid+1,r,lt,rt,del);
	Pushup(x);
}

int Query(int x,int l,int r,int lt,int rt) {
	if (lt<=l&&r<=rt) return f[x]*(m[x]<=2)+g[x]*(m[x]<=1);
	Pushdown(x);
	int mid=(l+r)>>1,ans=0;
	if (mid>=lt) ans+=Query(lch,l,mid,lt,rt);
	if (mid<rt) ans+=Query(rch,mid+1,r,lt,rt);
	return ans;
}

int main() {
	scanf("%d",&n);
	for (int i=1;i<=n;i++) {
		scanf("%d",&a[i]);
		pl[a[i]]=i;
	}
	Build(1,1,n);
	long long ans=0;
	for (int i=1;i<=n;i++) {
		int x=a[pl[i]-1],y=a[pl[i]+1];
		Updata(1,1,n,1,i,1);
		if (x&&x<i) Updata(1,1,n,1,x,-1);
		if (y&&y<i) Updata(1,1,n,1,y,-1);
		ans+=Query(1,1,n,1,i);
	}
	printf("%lld",ans-n);
	return 0;
}