传送门
题意:
给你两个长度为n的排列p,q,同时定义mex表示为在mex中,缺少的最小正整数:如
mex[1,3]=2,mex[1,2,3]=4。
问你有多少个l,r使mex([])=mex([])。
思路 :
可以从mex=1,2,3,4,..n这样一步一步看怎么找。
当mex为1的时候,就是找两边的1,然后分成的三部分计算答案就好
mex为2的时候,那么就是包括1的区间同时不能包括2,设2位置靠左边的下标为L,靠右边的下标为R。那么能够存在答案的只有三种情况:2的区间全在左边,全在右边,包括了1的区间。
1.全在左边(即R<l),那么就相当于左边l-r种情况和右边(n-r+1)种情况的排列组合,结果相乘即可。
2.全在右边(L>r),同理结果为(n-r+1)*(l-R).
3.包括了1,结果为(l-L)*(R-r),
之后区间进行叠加扩大。
当mex为3,4,5,...n的时候也是这样重复的进行计算,最后结果记得开ll。
代码:
#include<cstdio>
#include <iostream>
#include <algorithm>
#include <string.h>
#include <string>
#include <math.h>
#include<vector>
#include<queue>
#include<map>
#define sc_int(x) scanf("%d", &x)
#define sc_ll(x) scanf("%lld", &x)
#define pr_ll(x) printf("%lld", x)
#define pr_ll_n(x) printf("%lld\n", x)
#define pr_int_n(x) printf("%d\n", x)
#define ll long long
using namespace std;
const int N=1000000+100;
int n ,m,h;
int s[N];
void slove( )
{
int t;
sc_int(n);
map<int,int>q,w;
for(int i =1;i<=n;i++)
{
int x;
sc_int(x);
q[x]=i;
}
for(int i =1;i<=n;i++)
{
int x;
sc_int(x);
w[x]=i;
}
ll res=0;
ll l=q[1],r=w[1];
if(l>r)swap(l,r);
res+=(l-1)*l/2+(n-r)*(n-r+1)/2+max((ll)0,r-l-1)*(max((ll)0,r-l-1)+1)/2;
for(int i =2;i<=n;i++)
{
ll L=q[i],R=w[i];
if(L>R)swap(L,R);
if(L>r)res+=l*(L-r);
else if(R<l)res+=(n-r+1)*(l-R);
else if(L<l&&R>r) res+=(l-L)*(R-r);
l=min(l,L),r=max(r,R);
}
cout<<res+1;
cout<<endl;
}
int main()
{
int t;
slove();
return 0;
}