P5367 【模板】康托展开 / 计算数列在全排列中的排名

百度百科 - 康托展开

康托展开

$X=a_1\ast (n-1)!+a_2\ast (n-2)!+...+a_n\ast 0!$

其中，$a_i$ 表示给定整数序列的第i位在当前还未出现的元素中是排在第几个

可用于求当前序列在全排列中的排名 $=X+1$

逆康托展开

康托展开的实质是计算当前排列在所有由小到大全排列中的顺序，因此是可逆的

具体方法见百度百科 ↓

貌似线段树可以做

---

---

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e6 + 10;
const int mod = 998244353;
int n;

ll fac[N];
int bit[N];

void add(int x, int k) {
    for (; x <= n; x += x & -x) {
        bit[x] += k;
    }
}

int ask(int x) {
    int res = 0;
    for (; x; x -= x & -x) {
        res += bit[x];
    }
    return res;
}


int main() {
    ios::sync_with_stdio(0);

    cin >> n;
    fac[0] = 1;
    for (int i = 1; i <= n; ++i) {
        fac[i] = fac[i - 1] * i % mod;
        add(i, 1); //一开始全部数字都未使用过
    }
    ll ans = 0;
    for (int i = 1, x; i <= n; ++i) {
        cin >> x;
        ans = (ans + (ask(x) - 1) * fac[n - i]) % mod; //ask(x)求的是包含x在内的前缀和
        add(x, -1); //删去当前位置的数
    }
    cout << ans + 1 << endl;

    return 0;
}

---

板子

namespace CantorExpansion { // 康拓展开板子
    const int N = 20;
    ll fac[N];

    // 计算全排列 n!
    void initFac(int n) {
        fac[0] = 1;
        for (int i = 1; i <= n; i++) {
            fac[i] = fac[i - 1] * i;
        }
    }

    //康拓展开
    // res= cnt[1]*(n-1)! + cnt[2]*(n-2)! +...+ cnt[n]*0!
    // 根据原数组推断在全排列中的位置
    // vector<int> a - 原数组 由 0~n-1组成 (其实这里并不在意是1~n还是0~n-1 只要n个数是连续的就行)
    // n - 全排列范围 0~n-1
    ll Cantor(vector<int> a) {
        ll res = 0;

        int n = a.size();
        for (int i = 0; i < n - 1; i++) {
            int cnt = 0;
            //统计 已经遍历到第i个位置时 当前还没有出现的 比a[i]还小的数 的个数
            // 或者说 当i>j时 a[i]>a[j]的个数
            for (int j = i + 1; j < n; j++) {
                if (a[i] > a[j])
                    cnt++;
            }
            res += fac[(n - 1) - i] * cnt;
        }
        res++;//让排列序号从1开始
        return res;
    }

    // 逆康拓展开
    // 根据在全排列中的位置推断出数组
    vector<int> deCantor(ll x, int n) {
        //初始化vis数组
        vector<int> vis(n + 1, false);
        vector<int> res(n);
        int k;
        x--;//如果x的编号是从0开始的 就不需要这一步了 如果是1就得还原
        for (int i = 0; i < n; i++) {
            ll Min = x / fac[(n - 1) - i];//说明比a[i]小的数有Min个
            x = x % fac[(n - 1) - i];
            k = 0; // 如果全排列范围是1~n 修改这里 k=1
            for (int j = 0; j <= Min; k++) {
                if (!vis[k]) {
                    j++;
                }
            }
            k--;
            res[i] = k;
            vis[k] = true;
        }
        return res;
    }

    //输出排列
    void output(vector<int> a) {
        for (int i = 0, sz = a.size(); i < sz; i++) {
            printf("%d%c", a[i] , (i == sz - 1 ? '\n' : ' '));
        }
    }

}
using namespace CantorExpansion;