[学习笔记]康托展开

康托展开

一.什么是康托展开

​ 康托展开适用于计算一个排列结果在字典序全排列中的序号(很显然序号与这个排列形成一种双射关系),利用此算法可以构造排列的Hash表。

二.实现康托展开

​ 举个例,在集合({1,2,3})中选取三个数的全排列,问 3,2,1 是第几个。

对于这个问题,如果说让我用手来算的话,我会使用分类讨论比他小的排列,见下图。

至于累积的结果是否需要+1,取决于编号从0还是1开始。大致总结一下,我们可以得出康托展开的一般形式。

[ans=sum^{n}_{i=1}p_i*(n-i)! ]

其中(P_i)为在未选取的元素中比排列结果中的第i个数小的元素个数。在下文称之为系数。

三.求解系数

​ 很简单的,我们能过想到接近于(O(n^2))复杂度的暴力与vector求解,但是我们需要一个更高级的算法。

​ 如果说我们能用0,1表示某个元素是否存在,(我们使得初始集合中的元素成升序排列),那么对于前缀和而言,所代表的涵义就是比他小的元素个数,于是用此方法就可以快速求解系数。当这个数选取了之后,我们需要把它标记为不存在,前缀和涉及到修改,我们可以用树状数组求解。

四.Code(Luogu板子)

ll mod=998244353;
ll n,c[1000005];
void add(int x,int y){
	for(;x<=n;x+=x&(-x))c[x]+=y;
}
int sum(int x){
ll res=0;
while(x>0){
	res+=c[x];
	res%=mod;
	x-=x&(-x);
}	
return res-1;
}
ll now[1000005],ans,jc[1000005]={1};
void ready(){
	for(int i=1;i<=n;++i)jc[i]=(jc[i-1]*i)%mod;
}
int main(){
	cin>>n;
	ready();
	for(int i=1;i<=n;++i){
		cin>>now[i];
		add(i,1);
	}
	for(int i=1;i<=n;++i){
		ans=(ans+jc[n-i]*sum(now[i]))%mod;
		add(now[i],-1);
	}
	cout<<(ans+1)%mod;
	return 0;
}

五.逆康托展开

​ 在知道了康托展开的原理之后,我们其实可以通过序号求解排列(前文有提到他们是双射关系),于是考虑求解。

​ 我们可以知道一个东西(易证得)

[P_k(n-k)!>sum^{k-1}_{i=1}P_i(n-i)! ]

于是我们对每一个((n-i)!)进行取模与除就可以反向推出每一项的系数。得到系数以后,由于系数即代表大小关系,所以令(i=P_i+1),之后我们需要通过一个算法来求出第i小的元素,输出并删除,这里我用的是平衡树(FHQ)

挂一个已知系数求解元素的代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=50010;
int size_node[N],children[N][2],number_tree[N],number_heap[N],total,root, number_data_sets,order,x;
void update(int node_now) {size_node[node_now]=size_node[children[node_now][0]]+size_node[children[node_now][1]]+1;}
void split(int node_now,int k,int &tree_accept,int &tree_unaccept) {
    if(!node_now){
        tree_accept=tree_unaccept=0;
        return;
    }
    if(number_tree[node_now]>k) {
        tree_unaccept=node_now;
        split(children[node_now][0],k,tree_accept,children[node_now][0]);
    } else {
        tree_accept=node_now;
        split(children[node_now][1],k,children[node_now][1],tree_unaccept);
    }
    update(node_now);
}
int node_add(int k) {number_heap[++total]=rand(),number_tree[total]=k,size_node[total]=1;return total;}
int merge(int tree_one,int tree_two) {
    if(!tree_one||!tree_two)return tree_one+tree_two;
    if(number_heap[tree_one]<number_heap[tree_two]) {
        children[tree_one][1]=merge(children[tree_one][1],tree_two);
        update(tree_one);
        return tree_one;
    } else {
        children[tree_two][0]=merge(tree_one,children[tree_two][0]);
        update(tree_two);
        return tree_two;
    }
}
void Insert(int k) {
    int tree_one,tree_two;
    split(root,k,tree_one,tree_two);
	root=merge(merge(tree_one,node_add(k)),tree_two);
}
void extrack(int k) {
    int tree_less,tree_unless,tree_k;
    split(root,k,tree_less,tree_unless);
    split(tree_less,k-1,tree_less,tree_k);
 	tree_k=merge(children[tree_k][0],children[tree_k][1]);
    root=merge(tree_less,merge(tree_k,tree_unless));
}
int frank(int node_now,int k) {
    while(1) {
		if(size_node[children[node_now][0]]>=k)node_now=children[node_now][0];
        else if(size_node[children[node_now][0]]+1<k)k-=size_node[children[node_now][0]]+1,node_now=children[node_now][1];
        else return number_tree[node_now];
    }
}
ll k,n,s[N];
int main(){
	cin>>k;
	while(k--){
		total=0;
		scanf("%d",&n);
		for(int i=1;i<=n;++i){
			Insert(i);
			cin>>s[i];
		}
		for(int i=1;i<=n;++i){
			int u=frank(root,s[i]+1);
			printf("%d ",u);
			extrack(u);
		}
		printf("
");
	}
	return 0;
}
原文地址:https://www.cnblogs.com/clockwhite/p/12217947.html