洛谷【P1177】【模板】基数排序

题目传送门:https://www.luogu.org/problemnew/show/P1177

我对计数排序的理解:https://www.cnblogs.com/AKMer/p/9649032.html

所谓基数排序,我们可以简单的理解成是多关键字的计数排序。

假设我们要排序一些数据,把这些数据按照第一关键字\(>\)第二关键字\(>...>\)\(n\)关键字的优先级排序。

那么我们只需要按第\(n\)关键字,第\(n-1\)关键字...第\(1\)关键字倒着做计数排序就行了。我们做怎么样的计数排序呢?

假设\(k+1\)\(n\)号关键字已经做完了,我们来做第\(k\)个。由于之前的关键字已经全部从小到大排好了,那么我们在将第\(k\)个关键字做计数排序的时候只需要倒着枚举就可以了,假设某两个数据第\(k\)个关键字是相同的,那么显然是\(id\)越靠后的数据越大,这与计数排序的\(sum[a[i]]--\)有着十分好的契合度。至于第\(k\)个关键字不同的,自然也就会区分开来。

时间复杂度:\(O(n*关键字个数)\)

空间复杂度:\(O(n)\)

代码如下:

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int maxn=1e5+5;

int n;
int sum[10],id[maxn],rk[maxn];//id[i]表示第k+1到n号关键字排好后第i个数据的原本编号,rk[i]表示原本编号为i的数据在进行完k号关键字的计数排序后的排名

int read() {
    int x=0,f=1;char ch=getchar();
    for(;ch<'0'||ch>'9';ch=getchar())if(ch=='-')f=-1;
    for(;ch>='0'&&ch<='9';ch=getchar())x=x*10+ch-'0';
    return x*f;
}

struct num {
    int v[11];

    void print() {
        int pos=1;
        while(pos<=10&&!v[pos])pos++;//忽略前导零
        for(int i=pos;i<=10;i++)
            printf("%d",v[i]);
        printf(" ");
    }
}a[maxn];

int main() {
    n=read();
    for(int i=1;i<=n;i++) {
        int x=read();
        for(int j=10;j;j--)
            a[i].v[j]=x%10,x/=10;
        id[i]=i;//把整数拆成10个关键字排序,每个关键字的值域都是[0,9]
    }int *x=id,*y=rk;//因为到时候要交换数组,用指针就是O(1)的了。
    for(int key=10;key;key--) {//倒着做计数排序
        memset(sum,0,sizeof(sum));
        for(int i=1;i<=n;i++)sum[a[i].v[key]]++;
        for(int i=1;i<=9;i++)sum[i]+=sum[i-1];
        for(int i=n;i;i--)y[sum[a[x[i]].v[key]]--]=x[i];//y[i]记录排名第i的数据,记住不要把a[x[i]]写成a[i]了,x是之前全部排好后的顺序
        for(int i=1;i<=n;i++)x[y[i]]=i;//x[i]记录第i个数据的排名
        swap(x,y);//交换,不然做下一次id和rk的意思就反了,那么就会错
    }for(int i=1;i<=n;i++)
         a[x[i]].print();//输出
    return 0;
}
posted @ 2018-09-22 04:39  AKMer  阅读(412)  评论(0编辑  收藏  举报