前端时间看《编程珠玑》,其中的位图排序思路个人感觉还是比较巧妙,看完后自己用JAVA实现了一版,在此记录一下。
问题描述:对1亿个不重复的正整数(0-99999999)进行排序,内存只有50MB。
思路:充分利用计算机的每一个bit位来节约空间。如上需求中,我们预设1亿个bit位,初始置0。从待排序集合中每读取一个数,在对应的第多少位bit上置1。最后,遍历该bit位集合,每个为1的bit位,则输出对应的正整数,最终输出结果已经排序完成,所需内存空间仅需:1亿bit/8=12.5MB。
实现方案:使用int数组来模拟bit位序列。1 int = 4 byte = 32 bit,因此需要1亿/32=3125000个int数来表达1亿个bit位。
public class BitSort {
public static int length;
public static void main(String[] args) {
int count = 100000000;
int[] src = { 99999968, 2, 1, 5 };
// 一个integer占4个字节,32个bit,因此需要 N/32个整数串起来表示最大值为N的排序。
length = count >> 5;
if (count % 32 != 0) {
length += 1;
}
int[] a = new int[length];
for (int i = 0; i < src.length; i++) {
setNum(a, src[i]);
}
print(a);
}
public static void setNum(int[] a, int i) {
int index = i >> 5;
int mask = 1 << (i % 32);
int x = length - index - 1;
a[x] = a[x] | mask;
}
public static void print(int[] a) {
int length = a.length;
for (int i = 0; i < length; i++) {
int x = a[i];
for (int j = 31; j >= 0; j--) {
int mask = 1 << j;
if (mask == (x & mask)) {
System.out.print((length - i - 1) * 32 + j);
System.out.print(" ");
}
}
}
}
}