康托展开及其逆展开
康托展开
可以用于求 全排列 的排名(字典序)。
我们先给定一组排列。这里为了举例方便就拿了 OI-wiki 上的例子 \(\{2,5,3,4,1\}\) 了。
我们第一位是 \(2\) ,那么以 \(1\) 开头的就都比它小。后面的也是同理的搞法,就是看当前这位有多少个比它小还之前没有出现过的数(因为要保证是排列)。由于高位上的以及统计过了,我们只需要统计当前位下的即可。具体来说,多少个比它小还之前没有出现过的数的个数乘上目前长度 \(-1\) 的阶乘就是比它小的排列个数。
我们每位如此统计完加起来,然后再 \(+1\) (因为是排名)即可。这个过程我们用树状数组维护,复杂度 \(O(n\log n)\)。
逆康托展开
其实我们得知一个排列的排名后也可以得到其原本的排列。
我们先对于排名自减一次。然后我们每次对于长度 \(-1\) 的阶乘做除法,得到余数 \(r\) 和除数 \(x\) ,那么这一位上的数就应该是之前没出现过的数中的第 \(x+1\) 名,找到它并且把排名赋值为 \(r\) 进入子问题。
一直做长度次即可。这个过程我们用权值线段树维护可以做到 \(O(n\log n)\)。
总的参考代码:
#include
#define ll long long
#define db double
#define filein(a) freopen(#a".in","r",stdin)
#define fileot(a) freopen(#a".out","w",stdout)
#define sky fflush(stdout);
#define gc getchar
#define pc putchar
namespace IO{
inline bool blank(const char &c){
return c==' ' or c=='\n' or c=='\t' or c=='\r' or c==EOF;
}
inline void gs(char *s){
char ch=gc();
while(blank(ch) ) {ch=gc();}
while(!blank(ch) ) {*s++=ch;ch=gc();}
*s=0;
}
inline void gs(std::string &s){
char ch=gc();s+='#';
while(blank(ch) ) {ch=gc();}
while(!blank(ch) ) {s+=ch;ch=gc();}
}
inline void ps(char *s){
while(*s!=0) pc(*s++);
}
inline void ps(const std::string &s){
for(auto it:s)
if(it!='#') pc(it);
}
template
inline void read(T &s){
s=0;char ch=gc();bool f=0;
while(ch<'0'||'9'
inline void read(T &s,A &...a){
read(s);read(a...);
}
};
using IO::read;
using IO::gs;
using IO::ps;
const int N=1e6+3,mods=998244353;
inline int inc(int x,int y){
return (x+=y)>=mods?x-mods:x;
}
inline int dec(int x,int y){
return (x-=y)<0?x+mods:x;
}
inline int mul(int x,int y){
return 1ll*x*y%mods;
}
int n;
struct BitTree{
int c[N];
inline void add(int x,int k){
for(;x<=n;x+=(x&-x) ) c[x]+=k;
}
inline int find(int y){
int res=0;
for(;y;y-=(y&-y) ) res=inc(res,c[y]);
return res;
}
}t1;
struct SegTree{
struct node{
int l,r,mid;
int cnt;
}t[N<<2];
inline void pushup(int i){
t[i].cnt=t[i<<1].cnt+t[i<<1|1].cnt;
}
inline void build(int i,int l,int r){
t[i].l=l;t[i].r=r;t[i].mid=(l+r)>>1;
t[i].cnt=1;
if(l==r){
return;
}
build(i<<1,l,t[i].mid);
build(i<<1|1,t[i].mid+1,r);
pushup(i);
}
inline int kth(int i,int k){
if(t[i].l==t[i].r){
if(k!=1) return -1;
return t[i].l;
}
int key=t[i<<1].cnt;
if(key>=k) return kth(i<<1,k);
return kth(i<<1|1,k-key);
}
inline void modify(int i,int p,int k){
if(p<=t[i].l and t[i].r<=p){
t[i].cnt+=k;
return;
}
if(p<=t[i].mid) modify(i<<1,p,k);
if(t[i].mid+1<=p) modify(i<<1|1,p,k);
pushup(i);
}
}t2;
int fac[N];
inline int Contor(int *a){
memset(t1.c,0,sizeof(t1.c) );
int rk=0;
for(int i=1;i<=n;++i){
t1.add(a[i],1);
rk=inc(rk,mul(dec(a[i],t1.find(a[i]) ),fac[n-i]) );
}
return rk+1;
}
inline void InvContor(int x,int *a){
t2.build(1,1,n);--x;
for(int i=1;i<=n;++i){
int r=x/fac[n-i];
x%=fac[n-i];
a[i]=t2.kth(1,r+1);
t2.modify(1,a[i],-1);
}
}
int a[N],b[N];
int main(){
filein(a);fileot(a);
read(n);
fac[0]=fac[1]=1;
for(int i=2;i<=n;++i){
fac[i]=mul(fac[i-1],i);
}
for(int i=1;i<=n;++i){
read(a[i]);
}
int rk=Contor(a);
printf("%d\n",rk);
InvContor(rk,b);
for(int i=1;i<=n;++i){
printf("%d ",b[i]);
}pc('\n');
return 0;
}