算法通关村——位运算在查找重复元素中的妙用

在需要存储大量数字的场景下，相比直接用数组储存，位运算可以帮助减少储存的空间，例如这里有一道题：如何用4KB内存寻找重复元素

题目要求：给定一个数组，包含从1到N的整数，N最大为32000，数组可能还有重复值，且N的取值不定，若只有4KB的内存可用，该如何打印数组中所有重复元素。

这个时候我们就可以叫位运算帮忙了，位运算的原理是：因为一个整数有32位，我们可以用每一位来代表该数字是否存在，用1来代表存在，0则是不存在。这样子我们就可以用一个整数来表示32个数字的存在状态，最后代码如下：

public class FindDuplicatesIn32000 {
    public void checkDuplicates(int[] array) {
        BitSet bs = new BitSet(32000);
        for (int i = 0; i < array.length; i++) {
            int num = array[i];
            int num0 = num - 1; // 调整索引
            if (bs.get(num0)) {
                System.out.println(num); // 找到重复数字
            } else {
                bs.set(num0); // 存储数字
            }
        }
    }

    class BitSet {
        int[] bitset;

        public BitSet(int size) {
            bitset = new int[size >> 5]; // 使用位运算创建数组
        }

        boolean get(int pos) {
            int wordNumber = (pos >> 5); // 计算wordNumber
            int bitNumber = (pos & 0x1F); // 计算bitNumber
            return (bitset[wordNumber] & (1 << bitNumber)) != 0; // 查询位状态
        }

        void set(int pos) {
            int wordNumber = (pos >> 5); // 计算wordNumber
            int bitNumber = (pos & 0x1F); // 计算bitNumber
            bitset[wordNumber] |= 1 << bitNumber; // 设置位
        }
    }
}