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

传送门

题意：

给你两个长度为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;
}