题目要求:给定一个数组,包含从1到N的整数,N最大为32000,数组可能还有重复值,且N的取值不定,若只有4KB的内存可用,该如何打印数组中所有重复元素。
public class FindDuplicatesIn32000 {
public void checkDuplicates(int[] array) {
BitSet bs = new BitSet(320000); // 创建一个320000比特的位向量
for (int i = 0; i < array.length; i++) {
int num = array[i];
int num0 = num - 1;
if (bs.get(num0)) { // 如果位向量中对应位置已经为1,则说明出现了重复元素
System.out.println(num);
} else {
bs.set(num0); // 否则将对应位置设置为1
}
}
}
class BitSet {
int[] bitset;
public BitSet(int size) {
this.bitset = new int[size >> 5]; // 每个int有32比特,所以除以32得到需要的int数组长度
}
boolean get(int pos) {
int wordNumber = (pos >> 5); // 除以32,确定所在的int位置
int bitNumber = (pos & 0x1F); // 取余32,确定所在的比特位置
return (bitset[wordNumber] & (1 << bitNumber)) != 0; // 检查对应比特位置是否为1
}
void set(int pos) {
int wordNumber = (pos >> 5); // 除以32,确定所在的int位置
int bitNumber = (pos & 0x1F); // 取余32,确定所在的比特位置
bitset[wordNumber] |= 1 << bitNumber; // 将对应比特位置设置为1
}
}
}
在这里主要是利用位向量的思想,节省空间。
get(int pos)
方法用于获取指定位置的位值。它接受一个整数参数pos
,表示要获取的位的位置。首先,通右移5位计算出该位置所在的字(word)的索引wordNumber
。然后,通过对pos
进行位与运算(与0x1F相与),可以获取到该位置在该字中的位索引bitNumber
。最后,通过对bitset[wordNumber]
和(1 << bitNumber)
进行位与运算,判断该位置上的位是否为1,如果是则返回true
,否则返回false
。
set(int pos)
方法用于设置指定位置的位值。它也接受一个整数参数pos
,表示要设置的位的位置。首先,通过右移5位计算出该位置所在的字(word)的索引wordNumber
。然后,通过对pos
进行位与运算(与0x1F相与),可以获取到该位置在该字中的位索引bitNumber
。最后,通过对bitset[wordNumber]
和(1 << bitNumber)
进行位或运算,将该位置上的位设置为1。