前言
tableSizeFor方法其实就是获取了最接近你传入数值的2的N次方。
例如:传入18,返回32,传入9,返回16等。
那如果这样一个需求给到你,你会选择怎么实现呢?
我们来看一下HashMap里的源码,大佬是怎么实现的。
HashMap里的实现
static final int tableSizeFor(int cap) {//cap传入15
int n = cap - 1;
14
//移位运算
n |= n >>> 1;
n |= n >>> 2;
n |= n >>> 4;
n |= n >>> 8;
n |= n >>> 16;
return (n < 0) ? 1 : (n >= MAXIMUM_CAPACITY) ? MAXIMUM_CAPACITY : n + 1;
}
第一行很简单,为什么要-1放在最后说,最后一行是两个三目运算符,其中之一操作是n+1,都很容易理解。关键是中间五步移位加上或运算。
移位的思想
说一下我理解的作者的思想:
2的整数幂用二进制表示都是最高有效位为1,其余全是0,比如十进制8和32,下图只用了一个字节示意。
对任意十进制数转换为2的整数幂,结果是这个数本身的最高有效位的前一位变成1,最高有效位以及其后的位都变为0。
核心思想是,先将最高有效位以及其后的位都变为1,最后再+1,就进位到前一位变成1,其后所有的满2变0。所以关键是如何将最高有效位后面都变为1。
还是用图来示意。这里将十进制的25转换为32。
作者的做法是先移位,再或运算。
右移一位,再或运算,就有两位变为1;
右移两位,再或运算,就有四位变为1,,,
最后右移16位再或运算,保证32位的int类型整数最高有效位之后的位都能变为1.
全过程示意图
自己觉得理解了,但是感觉文章写出来很绕,估计看到这里的你也有这种感觉。这里对整个过程画图示意。
初始值
选取任意int类型数字,下图x表示不确定0或者1.
我们目的是将所有的x变为1,如下图
最后+1,就能进位得到2的整数幂。
我们要做的就是不断通过右移+或运算来达到目的。
右移一位+或运算
可以看出,右移一位再或运算,有两位变成了1。
右移二位+或运算
右移两位再或运算,有四位变成了1。
右移四位+或运算
右移四位再或运算,有八位变成了1。
右移八位+或运算
右移八位再或运算,有十六位变成了1。
右移十六位+或运算
右移十六位再或运算,注意这里不是三十二位全变,而是最高位后面的全变1。
结果+1
可以看出,不管x是多少,我们都能将其转换为1。而且分别经过1,2,4,8,16次转换,不管这个int类型值多大,我们都会将其转换,只是值较小时,可能多做几次无意义操作。
初始容量-1
之所以在开始移位前先将容量-1,是为了避免给定容量已经是8,16这样2的幂时,不减一直接移位会导致得到的结果比预期大。比如预期16得到应该是16,直接移位的话会得到32。在上图中就是所有x本身已经是0的情况下,不减1得到的结果变大了。
总结
tableSizeFor方法其实就是利用二进制里的位移,让二进制里每一位都变成1,这样再加上1,就能够获取到最近的2的N次幂,实在是精妙。