题目描述
两个整数的 汉明距离 指的是这两个数字的二进制数对应位不同的数量。
计算一个数组中,任意两个数之间汉明距离的总和。
- 示例:
输入: 4, 14, 2
输出: 6
解释: 在二进制表示中,4表示为0100,14表示为1110,2表示为0010。(这样表示是为了体现后四位之间关系)
所以答案为:
HammingDistance(4, 14) + HammingDistance(4, 2) + HammingDistance(14, 2) = 2 + 2 + 2 = 6.
注意:
数组中元素的范围为从 0到 109。
数组的长度不超过 104。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/total-hamming-distance
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
这道题很容易想到暴力法,利用双循环再结合【LeetCode】第461题——汉明距离(难度:简单)的解题思路,但是这种方法会超时,当然一个中等题也不会允许你通过暴力法解出。
所以需要换种思路,把循环层数减少一层。
大家可以考虑一下,对于长度为 n 的数组nums[],统计每个数字第 i 位的 0或1 的数量(此处以1为例)为 c,那么当前位的所有数字的汉明距离和为 c × ( n − c ) c×(n-c) c×(n−c),也就是01排列组合问题的答案
如此一来便减少了一层循环,直接上代码
代码详解
class Solution {
public int totalHammingDistance(int[] nums) {
int ans = 0;
for(int i = 0; i < 32; ++i) { // 因为题目给出数值限制,这里i也可以小于30
int countOne = 0; // 统计第i位1的个数
for(int j = 0; j < nums.length; ++j) {
if((nums[j] & 1) == 1) {
++countOne;
}
nums[j] = nums[j] >> 1; // 不断右移1位
}
ans += countOne * (nums.length - countOne); // 汉明距离累加
}
return ans;
}
}
注意点
- 逐位计算汉明距离也是厉害,这思路太强了,注意不要被此题的简单题的做法影响
- 友情链接:【LeetCode】第461题——汉明距离(难度:简单)