两个整数的 汉明距离 指的是这两个数字的二进制数对应位不同的数量。
计算一个数组中,任意两个数之间汉明距离的总和。
示例:
输入: 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
。
暴力超时法
class Solution {
public:
int totalHammingDistance(vector<int>& nums) {
int res = 0;
for(int i = 0; i < nums.size();i++){
for(int j = i; j < nums.size(); j++){
res += hammingDistance(nums[i], nums[j]);
}
}
return res;
}
int hammingDistance(int x, int y) {
int count = 0;
while(!(x== 0 && y == 0)){
if(x%2 != y%2){
count++;
}
// cout<<x<<" "<<y<<endl;
x = x /2;
y = y /2;
}
return count;
}
};
按位计算
class Solution {
public:
int totalHammingDistance(vector<int>& nums) {
int res = 0;
int len = nums.size();
if (len == 0)
return 0;
for (int i = 0; i < 32; ++i)
{
int pos = 0;
int neg = 0;
for (int j = 0; j < len; ++j)
{
if (nums[j] & 1)
++pos;
else
++neg;
nums[j] >>= 1;
}
res += (neg * pos);
}
return res;
}
};