JAVA的BIT数组

写个小东西，要去重复数字，用到BIT数组，虽然JAVA已经提供了一个BitSet，不过自己手痒，又写了一个简单的

原理就不写了，网上一大堆

 import java.util.Iterator;
 import java.util.function.BiConsumer;
 
 public class BitArray implements Iterable<Boolean>{
     //表示1<<n的值，提高效率，不用每次计算
     private final byte[] MASK = new byte[]{1,2,4,8,16,32,64,(byte)128};
     byte[] bits;
     int max = 0;
 
     /**
      * 构造一个Bit数组
      * @param max 最大位数
      */
     public BitArray(int max){
         this.max = max;
         int len = max / 8 + 1;
 
         bits = new byte[len];
     }
 
     /**
      * 设置第N位的值
      * @param index Bit索引
      * @param value 值
      */
     public void set(int index,boolean value){
         int i = index / 8;
         int move = index % 8;
 
         bits[i] = (byte)(bits[i] | MASK[move]);
     }
 
     /**
      * 取得第N位的值
      * @param index Bit索引
      * @return
      */
     public boolean get(int index){
         int i = index / 8;
         int move = index % 8;
 
         return (bits[i] & MASK[move]) == MASK[move];
     }
 
     /**
      * 显示所有位
      */
     public void show(){
         for(int i=0; i<bits.length; i++){
             byte b = bits[i];
             for(int bitIndex=0; bitIndex<8;bitIndex++){
                 System.out.print( ((b>>bitIndex) & 1) + " ");
             }
             System.out.println();
         }
     }
 
     /**
      * 提供遍历接口
      */
     public Iterator<Boolean> iterator() {
         return new Iterator<Boolean>(){
             private int i = 0;
 
             public boolean hasNext() {
                 return i <= max;
             }
 
             public Boolean next() {
                 return get(i++);
             }
 
         };
     }
 
     /**
      * 遍历，偷懒用了JAVA8的新接口
      * @param fun
      */
     public void forEach(BiConsumer<Integer,Boolean> fun){
         int total = 0;
         for(int i=0; i<bits.length; i++){
             byte b = bits[i];
             for(int bitIndex=0; bitIndex<8 && total<=max;bitIndex++,total++){
                 fun.accept(total, ((b>>bitIndex) & 1) == 1);
             }
         }
     }
 }

使用方式：

public static void main( String[] args ) throws Exception
{
    BitArray bits = new BitArray(18);
    bits.set(18,true);
    System.out.println("position 3 : " + bits.get(3));
    System.out.println("position 18 : " + bits.get(18));
    System.out.println("--------------------------");
 
    //遍历方式 一
    int i = 0;
    for(Boolean result : bits)
        System.out.println(i++ + " : " + result);
 
    System.out.println("--------------------------");
 
    //遍历方式二
    BiConsumer<Integer,Boolean> fun = (index, value)->{
        System.out.println(index + " : " + value);
    };
    bits.forEach(fun);
}

输出结果：

/*
position 3 : false
position 18 : true
--------------------------
0 : false
1 : false
2 : false
3 : false
4 : false
5 : false
6 : false
7 : false
8 : false
9 : false
10 : false
11 : false
12 : false
13 : false
14 : false
15 : false
16 : false
17 : false
18 : true
--------------------------
0 : false
1 : false
2 : false
3 : false
4 : false
5 : false
6 : false
7 : false
8 : false
9 : false
10 : false
11 : false
12 : false
13 : false
14 : false
15 : false
16 : false
17 : false
18 : true
*/