手记

20 亿个数字在 4G 内存中如何去重排序:快来试一试 BitMap

有一道流传广泛的面试题:
给你一台 4G 内存的机器,一组 20 亿个无序正整数,如何快速地判断一个正整数 N 是否在这组数字中?或者如何快速地对这组数据排重后排序?
让我们先算算 20 亿个整数会占用多大的内存空间,Java 的 int 类型占用 4 个字节,那么 20 亿 * 4 再换算成 G 大约是 7.5G,大于题目中 4G 内存的限制,无法一次性地放到内存中;
这时候有些伙伴会说:“把数据放到磁盘上,然后分批将数据读取到内存中就行查询”,但是这种方法会导致多次磁盘 IO,而且只能解决第一个查找的问题,排序就没有办法做到了。

BitMap 的概念

BitMap 能够很好地解决这个问题;它是用一个 Bit 位来标记某个元素对应的 Value, 而 Key 即是该元素,比如我们初始化一个类型为 bit、长度为 8 的数组,数组下标 0-7,数组中的内容 1 表示存在,0 表示不存在,那么:

00000001 下标为 0 的位置,对应值是1,那么表示 0;同理:
00000010 表示 1;
00000100 表示 2;
00001000 表示 3;

10000000 表示 7;

如果一组数据 {2,3,4,7} 放到同一个数组中的话,就是 10011100:

如果按照 int 数组存储,{2,3,4,7} 需要 4 * 4 * 8 个 bit 才能存储的数据,但是现在 BitMap 只需要 8 个 bit 就可以存储,很大地节省了存储空间,并且排重后的排序也变的非常简单了;如果用 byte 实现的话,只需要 1 个 byte 就可以(1 byte = 8 bits)。

如果增加了一个数字 10 呢,那么 1 个 byte 就不够了:

数据结构及初始化

我们可以得知,BitMap 的容量大小取决于最大的那个数值,比如要存储 {2,3,4,7,10}:

  • 如果用 bit 数组实现(假如有的话),那么需要 10 + 1 个长度;
  • 如果是用 byte 数组实现,那么需要 10/8 + 1 个长度;
  • 如果是用 int 数组实现,那么就需要 10/32 + 1 个长度(1 个 int 等于 4 个 bytes,等于 32 个 bits);

明白了这点之后,一个简单的 BitMap 数据结构也就可以确定了:

public class BitMap {
    //数据
    private byte[] bits; 
    //最大值
    private int max_value;
    //容量
    private int capacity;
    
    /**
     * 初始化
     * @param capacity
     */
    public BitMap(int max_value){
        this.max_value = max_value;
        //1bit存储8个数据,存储最大值为 max_value 的数组需要 max_value/8+1 个 byte,除以8就是右移3位
        this.capacity = (max_value >> 3 ) + 1;
        bits = new byte[capacity];
    }
}

添加数据

添加数据,需要快速地定位到这个元素要存到整个数组中的哪个位置,这里有两个概念:

  • 索引号 index:数据保存在整个数组的哪个下标中;
  • 位置号 position:数据在这个下标元素的哪个位置;

比如 10 保存在 index = 1,position = 2(从 0 开始) 这个位置中,经推算可得:

index = N / 8
position = N % 8

知道了 10 保存的位置之后,怎么把对应位置的数据更改成 1 呢?可以用“位或”运算。将 10 添加到 BitMap 中的完整步骤如下:

  • 计算 index = 10/8 = 1 ;
  • 计算 position = 10%8 = 2 ;
  • 将 byte[1] 的数据与 0000100 做“位或”运算,其中 0000100 是通过对 1 左移 2 得到。

完整的代码如下:

public void add(int num){
    //数据保存在整个数组的哪个下标中
    int index = num / 8;
    //数据在这个下标元素的哪个位置
    int position = num % 8;
    
    bits[index] |= 1<<position;
}

判断数字是否存在

知道了如何判断数字的索引号和位置号之后,判断数字是否存在也就容易了,直接使用“位与”运算,代码如下:

public boolean contains(int num){
  if(num > max_value){
    return false;
  }
  //数据保存在整个数组的哪个下标中
  int index = num / 8;
  //数据在这个下标元素的哪个位置
  int position = num % 8;
  return (bits[index] & 1<<position) != 0;
}

测试

让我们做一下测试吧:

public class BitMapTest {
  public static void main(String[] agrs){
    BitMap bm = new BitMap(100);
    
    bm.add(1);
    bm.add(12);
    bm.add(14);
    bm.add(51);
    bm.add(71);
    bm.add(100);
    
    System.out.println("12:" + (bm.contains(12)?"存在":"不存在"));
    System.out.println("13:" + (bm.contains(13)?"存在":"不存在"));
    System.out.println("51:" + (bm.contains(51)?"存在":"不存在"));
    System.out.println("66:" + (bm.contains(66)?"存在":"不存在"));
    System.out.println("100:" + (bm.contains(100)?"存在":"不存在"));
  }
}

运行结果:

12:存在
13:不存在
51:存在
66:不存在
100:存在

从结果可以看到,判断的都很准确,当然这只是一个最简单的BitMap实现,它还存在着很多问题,比如我们必须知道数据中最大的那个数字是多少,这个可以采用动态扩容的方式解决;
在 JDK 中,已经有对应实现的数据结构类 java.util.BitSet,我们可以不用强撸 BitMap,直接使用 BitSet 就好了,或者使用谷歌封装的 EWAHCompressedBitmap

优缺点

优点:

  • 占用内存空间低,可以极大地节约空间;
  • 运算效率高,查找、去重都不需要遍历全部数据;

缺点:

  • 所有的数据不能重复,相当于直接就是排重过的;
  • 如果数据只有两个:1 和 10000000,使用 BitMap 得不偿失,只有当数据比较密集时才有优势。

本章节介绍了 BitMap 的概念和基本实现,后续会介绍 BitMap 在实际开发中的应用。

0人推荐
随时随地看视频
慕课网APP