題目鏈接: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;
}