两个整数的 汉明距离 指的是这两个数字的二进制数对应位不同的数量。
计算一个数组中,任意两个数之间汉明距离的总和。
示例:
输入: 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),而如果知道规律可以把复杂度降到O(n),假如输入为:4,14,2
4 0100
14 1110
2 0010
我们只需要统计每列的1的个数和0的个数,相乘就能得到这一列的汉明距离。
最右端:1 0个,0 3个 汉明距离:0*3=0
次右端:1 2个,0 1个 汉明距离:2*1=2
。。。以此类推,最后加起来就行了。
参考代码:
class Solution {
public:
int totalHammingDistance(vector<int>& nums) {
int res = 0;
for (int i = 0; i < 32; i++) {
int cnt = 0;
for (auto num : nums) {
if (num&(1 << i)) cnt++;
}
res += ((nums.size() - cnt)*cnt);
}
return res;
}
};