题目链接:D. Misha and Permutations Summation
题意:两个全排列,算出这两个全排列分别是第几个全排列,然后这两个数字相加,取模(n!),然后再输出一个全排列表示第几个全排列,第几个是从0开始编号的。
大概在暑假的时候洛谷上看到一个变进制数的概念,就是将一个全排列转化为一个变进制数:yummy的博客
然后将两个变进制数相加,如果最高位还会产生进位说明相加超过了(n!),就相当于取模了。
然后再将变进制数转化为全排列。
看看lt的:ltnb
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e5+7;
int sum[maxn];
int n;
void add(int x,int y){ for(;x<=n;x+=x&-x) sum[x]+=y; }
int ask(int x){ if(x==0) return 0; int res=0; for(;x;x-=x&-x) res+=sum[x]; return res; }
int A[maxn],B[maxn];
int p[maxn];
int res[maxn];
bool check(int mid,int x){
return ask(mid)<=mid-x-1;
}
int S[maxn<<2|1];
void pushup(int k){S[k]=S[k<<1]+S[k<<1|1];}
int ask(int l,int r,int k,int x){
if(l==r) return l;
int mid=l+r>>1;
if(mid-l-S[k<<1]>=x) return ask(l,mid,k<<1,x);
return ask(mid+1,r,k<<1|1,x-(mid-l+1-S[k<<1]));
}
void upd(int l,int r,int k,int id){
if(l==r){
S[k]=1;
return ;
}
int mid=l+r>>1;
if(id<=mid) upd(l,mid,k<<1,id);
else upd(mid+1,r,k<<1|1,id);
pushup(k);
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;++i){
scanf("%d",&p[i]);
++p[i];
add(i,1);
}
for(int i=1;i<=n;++i){
A[i]=ask(p[i])-1;
add(p[i],-1);
}
memset(sum,0,sizeof(sum));
for(int i=1;i<=n;++i){
scanf("%d",&p[i]);
++p[i];
add(i,1);
}
for(int i=1;i<=n;++i){
B[i]=ask(p[i])-1;
add(p[i],-1);
}
//for(int i=1;i<=n;++i) cout<<A[i]<<" ";cout<<endl;
//for(int i=1;i<=n;++i) cout<<B[i]<<" ";cout<<endl;
int t=0;
for(int i=n,mod=1;i>=1;--i,++mod){
int h=t+A[i]+B[i];
p[i]=h%mod;
t=h/mod;
}
//for(int i=1;i<=n;++i) cout<<p[i]<<" ";cout<<endl;
for(int i=1;i<=n;++i){
int id=ask(1,n,1,p[i]);
upd(1,n,1,id);
res[i]=id;
}
for(int i=1;i<=n;++i) printf("%d ",res[i]-1);
return 0;
}