基础模板LV3_康托展开（LuoGu P5367）

康托展开 模板

题目描述

求 1∼N 的一个给定全排列在所有 1∼N 全排列中的排名。结果对 998244353 取模。

输入格式

第一行一个正整数 N。

第二行 N 个正整数，表示 1∼N 的一种全排列。

输出格式

一行一个非负整数，表示答案对 998244353 取模的值。

输入输出样例

输入 #1

3
2 1 3

输出 #1

3

输入 #2

4
1 2 4 3

输出 #2

2

说明/提示

对于10%数据，1≤N≤10。

对于50%数据，1≤N≤5000。

对于100%数据，1≤N≤1000000。

康托展开简介

康托展开是一个全排列到一个自然数的双射，常用于构建哈希表时的空间压缩。 康托展开的实质是计算当前排列在所有由小到大全排列中的顺序，因此是可逆的。百度百科

就而言之就是一个乱序的数列(1~N)可以用一个数值映射，实际上是全排列的第k个，然后找到这个k输出。

简单理解

模板从LV3开始有必要多讲一些，尤其是对于洛谷的毒瘤题，数据范围都非常大，一言不合就是10⁶级别，这个级别输入往往要用快读 （出题人就是想卡你常数 ，而且做法都需要得到优化，如果是对于50%数据的话，在知道康托展开的公式之后，就可以很容易得用暴力写出O(n²)的算法过掉这道题，然而他把数据放到1e6了之后就要用O(n)或者O(nlogn)算法，所以一般就要用上数据结构维护一些操作。

灵魂画图

Key1

首先看第一部分，通过数列1 3 2 4 算出最后的结果是2。
顺便提及一下他的公式，虽然他本不应该出现在这篇博客里。

其中, a_i为整数,并且
a_i表示原数的第i位在当前未出现的元素中是排在第几个

Key2

这里是他的伪代码，我们首先预处理1e6以内所有的阶乘，之后就可以用O(1)的时间查询到，然后我们需要处理的就是这个T，如果用暴力每一次就是O(n)，那么时间复杂度就会变成O(n²)，所以我们需要引入树状数组优化这个步骤。

Key3

在这里我们选择了用树状数组来维护这个操作，首先N <= 1e6，然后给1~n的数字都执行一次add(i, 1)，这类似与树状数组求第k大数字类似，如果是1的话表示该位置上存在一个数字，然后就可以在logN的时间内找到比x小并且此前没有出现过的数字的数量，需要注意的是，出现过的数字应当执行add(i, -1)就能够表示从集合中删去这个数字，因为在计算前缀和的时候就不会把这个数字计算进去了。

时间复杂度分析

O(n)时间预处理阶乘
O(nlogn)时间预处理树状数组

对于每一个数字而言都有
O(logn)查询比自己小又没有出现过的数字
O(1)查询阶乘

所以整体时间复杂度为O(nlogn)

参考代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N = 1e6 + 10, MOD = 998244353;
ll fact[N];
int a[N];
int n;
int tr[N];

int lowbit(int x) {
	return x & -x;
}

void add(int x, int k) {
	for (int i = x; i <= n; i += lowbit(i)) tr[i] += k;
}

int query(int x) {
	int res = 0;
	for (; x; x -= lowbit(x)) res += tr[x];
	return res;
}
void init() {
	fact[0] = 1;
	for (int i = 1; i <= n; i++) {
		fact[i] = fact[i - 1] * i % MOD;
	}
}

ll solve(){
	ll ans = 0;
	for (int i = 1; i <= n; i++) {
		int x = a[i]; // 这个数字
		int T = query(x) - 1; // 比这个数字小的数字没有被使用过的次数
		ans = (ans + T * fact[n - i]) % MOD;
		add(x, -1); // 删掉这个数字
	}
	return (ans + 1) % MOD;
}

int main() {
	scanf("%d", &n);
	init(); // 初始化
	for (int i = 1; i <= n; i++) scanf("%d", a + i);
	for (int i = 1; i <= n; i++) add(i, 1); // 树状数组数值初始化，相当于每个数最开始都存在
	cout << solve() << endl;
	return 0;
}