D. Moscow Gorillas Codeforces Round 852 (Div. 2)

 

题意：

是给定两个排列p和q，求出子序列 [l, r] 的个数，使得 MEX(a[l...r]) = MEX(b[l...r])。此处 MEX定义为序列中未出现的最小正整数。

思路：首先记录一下所有元素的下标，MEX = 1时，假设p中1的位置为L，q中1的位置为R (L<=R)，此时需要将L,R这两个点挖掉（如果1在区间内MEX就不等于1了），把整段分成三部分，满足条件的区间一定时同时存在于三段中的其中一段，如1 <= l <= r <= L-1。

显然结果应该是：cal(l-1)+cal(max(0ll,r-l-1))+cal(n-r)，此处的cal定义为（x*(x+1))/2;

接着考虑MEX = 2的情况。MEX = 2时，说明区间里一定包含1，但不含2，那么2的位置就不能出现在【L,R】之间。设x为序列p中2的位置，y为序列q中2的位置，x<=y, x,y要么同时出现在【1,L-1】一侧，要么同时出现在【R+1,n】一侧，要么一边在【1,L-1】一边在【R+1,n】。

当MEX>=2时满足以下通式：

        1：如果pos[i]<minp[i-1].满足mex(p[l~r])=i的l,r，要满足pos[i]<l<=minp[i-1],maxp[i-1]<=r<=n

        2:如果pos[i]>minp[i-1].满足mex(p[l~r])=i的l,r，要满足1<=l<=minp[i-1],maxp[i-1]<=r<=n

        3:否则就是maxp[i-1]>=pos[i]>=minp[i-1],没有区间满足mex(p[l~r])=i;

 

另外一种解法可以看我hxd的题解：D. Moscow Gorillas_hide_on-BUSh的博客-CSDN博客

AC代码：

#include<bits/stdc++.h>
#define lmw ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
using namespace std;
const int N=1e6+10;
int a[N],f[N],s[N],b[N];
#define int long long
int cal(int x){
	return x*(x+1ll)/2;
}
signed main(){
	lmw;
	int n;
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		f[a[i]]=i;
	}
	for(int i=1;i<=n;i++){
		cin>>b[i];
		s[b[i]]=i;
	}
	int res=0;
	int l=f[1],r=s[1];
	if(l>r) swap(l,r);
	res+=cal(l-1)+cal(max(0ll,r-l-1))+cal(n-r);
	for(int i=2;i<=n;i++){
		int x=f[i],y=s[i] ;
		if(x>y) swap(x,y) ;
		if(y<l) res+=(l-y)*(n-r+1ll);
		else if(x>r) res+=l*(x-r);
		else if(x<l&&y>r) res+=(l-x)*(y-r);
		l=min(l,x);
		r=max(r,y);
	}
	cout<<res+1<<'\n';
}