java实现一个bitmap--《编程珠玑ch1》

/**
 * 用于排序一个满足以下条件的文件:
 *      (1) 文件中只范围在0～1024*1024*8-1的整数
 *      (2) 每个数字只出现1次或0次
 * 使用到的数组大小:
 *      约1MB
 * 排序时间复杂度:
 *      O(n), n为文件中出现的数字总数
 */
public class BitMap {
    public static void main(String[] args){
        // 该位图大小约为1MB, 实际会比1MB大, 因为对象头还会占一些字节
        int[] bitmap = new int[256*1024];
        // 该数组中一共有多少个bit位就能表示多大的数
        int total = 32 * 256 * 1024;

        // 从0-total中选几个数插入到bitmap中, 这里可以替换为从文件读取
        for (int i=0;i<total;i++){
            // 每1024个就插入一个数, 当然也可以每512个或者完全随机(但需要保证每个数字最多出现一次)
            if ((i & 1023) == 0){
                put(bitmap, i);
            }
        }

        // 遍历位图, 输出被插入的数, 相当于做排序
        int base;
        int count;
        for (int i=0;i<bitmap.length;i++){
            count = 0;
            base = Integer.MIN_VALUE;
            while (base != 0){
                if ((bitmap[i] & base) != 0){
                    int r = (i<<5)+count;
                    System.out.println(r);
                }
                count++;
                base >>>= 1;
            }
        }

    }

    // 将input插入到bitmap中, 即修改对应bit位为1
    public static void put(int[] bitmap, int input){
        int index = input >> 5;
        int bit_index = input & 31;
        int base = Integer.MIN_VALUE;
        bitmap[index] |= (base >>> bit_index);
    }
}