统计汉明距离

题目描述

两个整数的 汉明距离 指的是这两个数字的二进制数对应位不同的数量。

计算一个数组中，任意两个数之间汉明距离的总和。

示例:

输入: 4, 14, 2

输出: 6

解释: 在二进制表示中，4表示为0100，14表示为1110，2表示为0010。（这样表示是为了体现后四位之间关系）
所以答案为：
HammingDistance(4, 14) + HammingDistance(4, 2) + HammingDistance(14, 2) = 2 + 2 + 2 = 6.

注意:

数组中元素的范围为从 0到 10^9。
数组的长度不超过 10^4。

题目分析

按照正常的思路暴力解的话，肯定是双重循环，比较每一个数字与其他未比较过的数字的汉明距离，最后求总和。这样子的话时间效率将会是O（n^2)。然而我们学习算法，当然是追求优化~

下面介绍一种优化的算法：

在计算汉明距离时，我们考虑的是同一比特位上的值是否不同，而不同比特位之间是互不影响的。

对于数组 nums中的某个元素 val，若其二进制的第 i 位为 1，我们只需统计 nums中有多少元素的第 i 位为 0，即计算出了 val 与其他元素在第 i 位上的汉明距离之和。

具体地，若长度为 n 的数组 nums 的所有元素二进制的第 i 位共有 c 个 1，n−c 个 0，则此数组在二进制的第 i 位上的汉明距离之和为 c⋅(n−c)

我们可以从二进制的最低位到最高位，逐位统计汉明距离。将每一位上得到的汉明距离累加即为答案。

具体实现时，对于整数 val 二进制的第 i 位，我们可以用代码 (val >> i) & 1 来取出其第 i+1 位的值。此外，由于 由于 $10^9$<$2^{30}$-1 ，意味着第我们可以直接从二进制从右往左数的的第 1 位枚举到第 29 位，第30位肯定为0。

正解代码

class Solution {
    public int totalHammingDistance(int[] nums) {
        int ans = 0, n = nums.length;
        // 从二进制的1位枚举到第29位
        for (int i = 0; i < 30; ++i) {
            int c = 0;
            // 统计 nums 中有多少元素的第 i位为 1
            for (int val : nums) {
                c += (val >> i) & 1;
            }
            // 累加每个位上得到的汉明距离
            ans += c * (n - c);
        }
        return ans;
    }
}

>>是带符号右移， 1&1 = 1 , x & 0 =0

这样子优化后

• 时间复杂度：O(n⋅L)。其中 n 是数组 nums的长度，L=30

• 空间复杂度仍然为：O(1)

来源：力扣（LeetCode）

。