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;
}

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