F. MEX vs MED （2000‘ 数学）

Codeforces Round #828 (Div. 3) F. MEX vs MED

原题链接

time limit per test 1 second

memory limit per test 256 megabytes

题面

You are given a permutation $p_1,p_2,\dots ,p_n$ of length $n$ of numbers $0,\dots ,n-1$. Count the number of subsegments $1\le l\le r\le n$ of this permutation such that $mex(p_l,p_{l+1},\dots ,p_r)>med(p_l,p_{l+1},\dots ,p_r)$.

$mex$ of $S$ is the smallest non-negative integer that does not occur in $S$. For example:

• $mex(0,1,2,3)=4$
• $mex(0,4,1,3)=2$
• $mex(5,4,0,1,2)=3$

$med$ of the set $S$ is the median of the set, i.e. the element that, after sorting the elements in non-decreasing order, will be at position number $ \left \lfloor \dfrac{|S|+1}{2}\right\rfloor$ (array elements are numbered starting from 1 and here $\lfloor v\rfloor$denotes rounding $v$ down.). For example:

• $med(0,1,2,3)=1$
• $med(0,4,1,3)=1$
• $med(5,4,0,1,2)=2$

A sequence of nn numbers is called a permutation if it contains all the numbers from $0$ to $n-1$ exactly once.

翻译

给定一个 $\mathrm{0...}n-1$ 的排列，求有多少个区间 $[l,r]$ 满足 $mex(p_l...p_r)>med(p_l...p_r)$。

其中$mex$的含义为区间中没有出现过的最小非负数 ；$med$ 的含义为区间中排名不超过一半的最大数字。

题解

$x$ 是某个区间的 $mex$ 的条件：该区间中 $\mathrm{0...}x-1$ 都出现了，但是 $x$ 没有出现。

我们记 $l,r$ 分别为 $\mathrm{0...}x-1$ 的所有数字在排列中出现的最靠前和最靠后的位置，记 $x$ 出现的位置为 $pos$。

注意到一个性质：

若 $l,r$ 在 $x$ 的同侧，则 $x$ 为某些区间的 $mex$；否则，$x$ 一定不可能是某些区间的 $mex$。

简要说明：

若 $l,r$ 在 $x$ 的同侧，则$[l,r]$ 中出现了小于 $x$ 的所有数字，但是没有出现 $x$， 所以 $[l,r]$ 的 $mex$ 值为 $x$ ;反之如果 $l,r$ 在 $x$ 的两侧，则任何一个区间都不可能同时包含小于 $x$ 的所有数字却不包含 $x$。

注意到第二个性质：

满足题意要求的 $mex>med$ 的区间，区间长度 $len$ 一定不超过 $2\times mex$

简要说明：

区间中一定包含 $\mathrm{0...}mex-1$ 已经占用了 $mex$ 个位置，而剩余的元素一定都是大于 $mex$ 的。如果剩余的元素多于 $mex$ 个，那么排名居中的数字一定会大于 $mex$

进而有了第三个性质：

如果一个区间长度为 $len$ ，并且包含了 $\mathrm{0...}\lfloor \frac{len-1}{2}\rfloor$ 的全部数字，那么这个区间一定是满足要求的。相反如果这个区间没有包含 $\mathrm{1...}\lfloor \frac{len-1}{2}\rfloor$ 中的某个数字， 那么这个区间一定是不满足要求的。

证明：

如果包含了 $\mathrm{0...}\lfloor \frac{len-1}{2}\rfloor$ 的所有数字，那么该区间的 $mex$ 至少为区间中第 $\lfloor \frac{len-1}{2}\rfloor +2$ 大的数字。而该区间的 $med$ 为第$\lfloor \frac{len+1}{2}\rfloor$ 大的数字。

由于$\lfloor \frac{len-1}{2}\rfloor +2>\lfloor \frac{len-1}{2}\rfloor$，因此一定有 $mex>med$。

反之同理。

由第三个性质就直接得到了解法：

从小到大枚举 $len$ ，并不断更新包含$\mathrm{0...}\lfloor \frac{len-1}{2}\rfloor$所有数字的左右边界 $l,r$。如果这个区间长度大于 $len$ 那么该长度下无解，否则该长度下包含区间 $l,r$ 的所有区间都是满足要求的。

另外，在更新 $l,r$ 的过程中还有一点小细节，见代码。

#include<bits/stdc++.h>
using namespace std;

#define int long long

/******************************************************************/
/******************************************************************/
/**************************以下是做题代码***************************/

const int N = 5e5 + 7;
int  pos[N], n, x, l, r, res;

void SolveTest() {
    cin >> n;
    for (int i = 1; i <= n; i++) {
        cin >> x;
        pos[x] = i;
    }
    l = r = pos[0];
    res = 1;//长度为1的区间只有单个的0满足要求

    //枚举区间的长度
    for (int len = 2; len <= n; len++) {

        // me为该区间长度下，满足要求的最小的mex值
        int me = (len - 1) / 2;

        //所有满足条件的区间：长度为len，包含子区间[l,r]
        l = min({l, pos[me], n + 1 - len});
        r = max({r, pos[me], len});
        res += max(0ll, len - (r - l));
    }
    cout << res << '\n';
}

/**************************以上是做题代码***************************/
/******************************************************************/
/******************************************************************/

signed main() {
#ifdef CQLOCAL
    freopen("C:/Users/20824/Desktop/C++代码/in.txt", "r", stdin);
    freopen("C:/Users/20824/Desktop/C++代码/out.txt", "w", stdout);
#endif
    int TestCase = 1;
    cin >> TestCase;
    for (int i = 1; i <= TestCase; i++) {
        SolveTest();
    }
    return 0;
}