计数排序
在某些特殊情况下,对数据进行排序,使用计数排序可以实现O(N)的时间复杂度。比如现在有K个数a0,a1,a2........ak,这些数的范围是0<=ai<N,这时候可以在代码中申请一个数组作为计数器,比如ctr[ai]用于统计ai的出现次数,最后根据ctr输出排序后的数据,简单代码如下:
int ctr[MAXN];//计数器
int data[MAXN];//存储未排序的数据
int n;//data数组中数据的个数
void CountingSort() {
int i,j;
memset(ctr, ctr+MAXN, 0);
for (i=0;i<n;i++)
ctr[data[i]]++;
for (i=0;i<MAXN;i++) {
for (j=0;j<ctr[i];j++)
printf("%d ", i);
}
}
位运算
通过计数排序可以降低排序的时间复杂度,但是对空间的要求比较高。假设未排序的数组中,每个元素最多出现一次,这样子就只需要用一个bit来作标记,而不需要一个int(32bits),示例代码如下:
#include<stdio.h>
#define BITPERWORD 32
#define SHIFT 5
#define MASK 0x1F
#define N 10000000
int bit[1+N/BITPERWORD];
void set(int i) {
bit[i>>SHIFT] |= (1<<(i & MASK));
}
void clear(int i) {
bit[i>>SHIFT] &= ~(1<<(i & MASK));
}
int getBit(int i) {
return bit[i>>SHIFT] & (1<<(i & MASK));
}
int main() {
int i;
// printf("%d\n", MASK);
for (i = 0; i < N; i++)
clear(i);
while (1) {
scanf("%d", &i);
if (i == -1)
break;
printf("%d %d\n", i >> SHIFT, (1<<(i & MASK)));
set(i);
}
for (i = 0; i < N; i++) {
if (getBit(i))
printf("%d\n", i);
}
return 0;
}