D. Misha and Permutations Summation(变进制数康拓展开)

题目链接: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;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章