深入剖析 Java 中的 tableSizeFor 方法：通过位运算计算最小的 2 的幂

知行小栈

于 2024-09-21 10:00:00 发布

阅读量899

点赞数 11

CC 4.0 BY-SA版权

分类专栏： Java 高级特性篇文章标签： java 开发语言

本文链接：https://blog.youkuaiyun.com/LearnerDL/article/details/142393274

Java 高级特性篇专栏收录该内容

12 篇文章

订阅专栏

前言

在 Java 的 HashMap 实现中，有一个常用的内部方法 tableSizeFor(int c)，它用于计算大于或等于指定整数 c 的最小 2 的幂。这在调整哈希表的容量时尤为重要，保证数组的容量总是 2 的幂次，能使哈希函数运算更加高效。

本文将通过详细分析 tableSizeFor 的源代码，解释其背后的位运算技巧和算法原理，并特别关注其中的一些关键细节，比如为什么位扩展到 16 位时停止。

`tableSizeFor` 方法的实现

首先，我们来看 tableSizeFor 方法的源码，它存在于 Java 的 HashMap 类中：

private static final int tableSizeFor(int c) {
    int n = c - 1;
    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;
}

该方法通过位操作来快速计算大于或等于输入值 c 的最小 2 的幂。它的执行效率很高，因为完全依赖于位操作，而非循环或条件判断。接下来，我们逐步解释这段代码的工作原理。

核心算法步骤解析

处理输入值

int n = c - 1;

这里，输入的值 c 被减去 1，这是为了应对输入值刚好是 2 的幂的情况。

举个例子，如果 c 正好是 2 的幂（比如 8、16、32 等），我们需要确保返回的结果还是 c 而不是更大的下一个 2 的幂。

示例：

如果 c = 8，c - 1 = 7（二进制表示为 0111）。
如果 c = 10，c - 1 = 9（二进制表示为 1001）。

位扩展操作

接下来是一系列位扩展操作，通过右移（>>>）和按位或（|）逐步将 n 的最高位 1 扩展到右边所有的位：

n |= n >>> 1;
n |= n >>> 2;
n |= n >>> 4;
n |= n >>> 8;
n |= n >>> 16;

这些位操作步骤的作用是将 n 的最高有效位 1 扩展到所有更低的位。我们来逐步解析其具体含义。

假设 c = 10，因此 n = 9（二进制 1001）。经过这些步骤后：

n |= n >>> 1：n 变成 1101，最高的 1 位被扩展到了它右边的 1 位。
n |= n >>> 2：n 变成 1111，最高两位的 1 扩展到了右边所有位。
后面的位移操作对结果已经没有影响，因为此时所有的低位都已经被填充为 1。

为什么进行位扩展

位扩展的目的是将任意整数 n 转换为一个“全 1”的二进制数，即最高位 1 以及它右边的所有位都被设置为 1。这样我们只需在最后一步加 1，就可以得到大于或等于输入值的最小 2 的幂。

例如，1001 经过逐步扩展变成 1111，然后加 1 得到 10000（16），这正是大于 10 的最小 2 的幂。

处理边界条件并返回结果

return (n < 0) ? 1 : (n >= MAXIMUM_CAPACITY) ? MAXIMUM_CAPACITY : n + 1;

如果 n 小于 0，返回 1，这是为了防止输入值过小的情况，确保最小值为 1。
如果 n 大于等于 MAXIMUM_CAPACITY，则返回 MAXIMUM_CAPACITY，这是为了防止数组的大小超过哈希表的最大容量。
否则，返回 n + 1，即大于等于 c 的最小 2 的幂。

为什么位扩展到 16 位时就停止了

一个值得注意的细节是，扩展操作最后一行是 n |= n >>> 16，也就是说，位扩展到 16 位时就停止了。那么，为什么在这个时候停止了呢？

32 位整数的限制

在 Java 中，int 类型是一个 32 位的有符号整数，占用 4 字节。因此，无论输入的数字是多少，最多只能用 32 位的二进制数来表示。

为了确保我们扩展到所有可能的位，我们只需要处理最多 32 位。在代码中，通过右移操作逐步扩展位数：

n >>> 1：扩展最高 16 位。
n >>> 2：扩展最高 8 位。
n >>> 4：扩展最高 4 位。
n >>> 8：扩展最高 2 位。
n >>> 16：扩展最高 1 位。

通过这 5 步操作，整个 32 位整数的所有位都可以被最高位的 1 覆盖。如果再向右移超过 16 位，就没有必要了，因为 32 位整数的所有位已经被扩展完毕。

其他类型的位扩展

如果我们使用的是 64 位的整数（long 类型），那么类似的操作就会继续扩展到 n >>> 32。但是在 32 位的 int 类型中，扩展到 n >>> 16 就足够覆盖所有位了。

示例解析

让我们通过几个示例进一步理解这个算法的工作过程：

输入 c = 5：

c - 1 = 4（二进制 0100）。
位扩展操作后，n = 0111。
n + 1 = 8（二进制 1000），返回 8。

输入 c = 17：

c - 1 = 16（二进制 10000）。
位扩展操作后，n = 11111。
n + 1 = 32，返回 32。

这些示例展示了 tableSizeFor 的实际效果：它能快速计算出大于等于 c 的最小 2 的幂，避免了循环或条件判断，使得运行时间是常数时间 O(1)。

为什么使用 2 的幂作为哈希表的容量

在哈希表（如 HashMap）中，底层数据结构通常是一个数组，而数组的长度最好是 2 的幂次。这是因为哈希表通常会使用位运算来计算哈希值的索引位置，而位运算（如 & 操作）比取模运算（%）要快得多。

假设数组长度是 2 的幂次，我们可以通过 h & (n - 1) 来高效地计算元素在数组中的索引，而无需使用相对昂贵的取模操作。这也是 tableSizeFor 方法的重要性所在，它帮助动态扩展哈希表时，始终保持容量为 2 的幂次，从而提升性能。

结语

通过本文的详细解析，我们深入了解了 Java 中 tableSizeFor 方法的实现原理。这个方法利用了巧妙的位操作，能够在常数时间内计算出大于等于给定数值的最小 2 的幂。特别是，位扩展的步骤精心设计到 16 位就停止，是为了匹配 32 位整数的限制，使得算法在效率和正确性上得到了保证。

这个方法广泛应用于哈希表（如 HashMap）中，确保数组容量始终是 2 的幂次，从而能通过位运算快速计算元素的索引，提升哈希表的性能。

位运算是编程中的一项重要技能，特别是在需要优化性能的场景下，像 tableSizeFor 这样的位操作技巧可以大大提高代码的运行效率。