题目描述
两个整数的 汉明距离 指的是这两个数字的二进制数对应位不同的数量。
计算一个数组中,任意两个数之间汉明距离的总和。
示例:
输入: 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 位的值。此外,由于 由于 1 0 9 10^{9} 109< 2 30 2^{30} 230-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)
。