Codeforces Round #779 (Div. 2) C. Shinju and the Lost Permutation

题面

题意

• 对于全排列 $p$，我们有以下内容。

  • $p$ 的前缀最大值 $b$

• 一个操作 “右移” ，比如 $[5,3,4,1,2]$ 右移会变成 $[2,5,3,4,1]$ (把最后一个数放到前面)

• 现在给定一个长度为 $n\le 10^5$ 的数组 $a$ ，问是否存在一个全排列 $p$ 满足:

  对于 $1\le i\le n$，$p$ 经过 $i$ 次右移后，其前缀最大值 $b$ 中不同的数有 $a_i$ 个。

  只输出 YES 或者 NO

思路

1. 旋转数组，使得 $a_i=1$ 在最前面。比如 $[5,3,4,1,2]\Rightarrow [1,2,5,3,4]$。
2. 注意到，每次平移，最多只会递增 1，于是 $a_{i+1}-a_i\ge 2$ 是非法的。$a_i=1$ 只有一个才是合法的。把整个过程想象为构造一个以 $b_i$ 为关键字的最大堆。把它对应的 $\max b_i$ 作为树的根，把结点距离树的距离 $d+1$ 作为 $a_i$。如果 $a_i<a_{i+1}$，$b_i$ 下扩展一个叶子结点 $b_{i+1}$，树深++。否则，找到一个 $d=a_i$ 的父亲节点，把当前结点接在它下面。只要递增<=1或者递减>=2，重复上述操作，一定可以把树填满。

代码

int n, a[N], b[N];

int main() {
    int T; cin >> T;
    while(T--) {
        cin >> n;
        rep(i, 1, n) cin >> a[i];
        int pos = 0;
        bool flag = true;
        rep(i, 1, n) {
            if(a[i] == 1) {
                if(!pos) pos = i;
                else flag = false;
            }
        }
        if(!pos) flag = false;
        rep(i, 1, n) {
            int p = (pos + i - 1) % n;
            if(!p) p = n;
            b[i] = a[p];
        }
        rep(i, 2, n) if(b[i] - b[i - 1] >= 2) flag = false;
        puts(flag ? "YES" : "NO");
    }
    return 0;
}