康托展开

最新推荐文章于 2024-09-19 22:05:37 发布

_Shmily

最新推荐文章于 2024-09-19 22:05:37 发布

阅读量128

点赞数

分类专栏：数论基本算法

本文链接：https://blog.csdn.net/zhaoxinxin1234/article/details/104228760

版权

数论基本算法专栏收录该内容

67 篇文章 0 订阅

订阅专栏

一、康托展开：
P5367 【模板】康托展开
树状数组维护。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#include<set>
#include<vector>
#define ll long long
#define llu unsigned ll
using namespace std;
const int mod=998244353;
const int maxn=1000100;
ll a[maxn],sum[maxn],fac[maxn];
int n;

void add(int x,int val)
{
    for(;x<=n;x+=(x&-x))
        sum[x]+=val;
}

int ask(int x)
{
    int ans=0;
    for(;x;x-=(x&-x))
        ans+=sum[x];
    return ans;
}

int main(void)
{
    scanf("%d",&n);
    fac[0]=1;
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&a[i]);
        fac[i]=fac[i-1]*i%mod;
    }
    ll ans=0;
    for(int i=1;i<=n;i++)
    {
        add(a[i],1);
        ans=(ans+(a[i]-ask(a[i]))*fac[n-i])%mod;
    }
    printf("%lld\n",ans+1);
    return 0;
}

二、康托展开逆运算：
洛谷：
UVA11525 Permutation
①、权值线段树维护：
70ms

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#include<set>
#include<vector>
#define ll long long
#define llu unsigned ll
using namespace std;
const int maxn=50500;
int n,x;
struct node
{
    int l,r,sum;
}t[maxn<<2];
void build(int l,int r,int cnt)
{
    t[cnt].l=l,t[cnt].r=r;
    t[cnt].sum=r-l+1;
    if(l==r) return ;
    int mid=(l+r)>>1;
    build(l,mid,cnt<<1);
    build(mid+1,r,cnt<<1|1);
}

int ask(int sum,int cnt)
{
    t[cnt].sum--;
    if(t[cnt].l==t[cnt].r) return t[cnt].l;
    if(sum<=t[cnt<<1].sum) return ask(sum,cnt<<1);
    else return ask(sum-t[cnt<<1].sum,cnt<<1|1);
}

int main(void)
{
    int tt;
    scanf("%d",&tt);
    while(tt--)
    {
        scanf("%d",&n);
        build(1,n,1);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&x);
            if(i!=1) putchar(' ');
            printf("%d",ask(x+1,1));
        }
        putchar('\n');
    }
    return 0;
}

②、树状数组倍增：

60ms

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#include<set>
#include<vector>
#define ll long long
#define llu unsigned ll
using namespace std;
const int maxn=50500;
int n,x,t,c[maxn];

void add(int x,int val)
{
    for(;x<=n;x+=(x&-x))
        c[x]+=val;
}

void init(void)
{
    for(int i=1;i<=n;i++)
        add(i,1);
    t=log2(n);
}

int ask(int x)
{
    int sum=0,now=0;
    for(int i=t;i>=0;i--)
    {
        if(now+(1<<i)<=n&&sum+c[now+(1<<i)]<x)//这里一定要用小于号，不然会出问题
            sum+=c[now+(1<<i)],now+=(1<<i);
    }
    return now+1;//然后返回的时候+1
}

int main(void)
{
    int tt;
    scanf("%d",&tt);
    while(tt--)
    {
        scanf("%d",&n);
        init();
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&x);

            if(i!=1) putchar(' ');
            int now=ask(x+1);
            printf("%d",now);

            add(now,-1);
        }
        putchar('\n');
    }
    return 0;
}