计数排序+位运算

计数排序

在某些特殊情况下，对数据进行排序，使用计数排序可以实现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;
}