leetcode-477. 汉明距离总和
两个整数的 汉明距离 指的是这两个数字的二进制数对应位不同的数量。
计算一个数组中,任意两个数之间汉明距离的总和。
示例:
输入: 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*n)
空间复杂度:O(1)
class Solution {
int Hamming(int x,int y)
{
int ans=0,ret=x^y;
while(ret)
{
if(ret%2==1) ans++;
ret/=2;
}
return ans;
}
public:
int totalHammingDistance(vector<int>& nums) {
int answer=0;
for(int i=0;i<nums.size();i++)
{
for(int j=i+1;j<nums.size();j++)
{
answer+=Hamming(nums[i],nums[j]);
}
}
return answer;
}
};
不对数字进行两两枚举,而是对二进制的每个位进行枚举,由于数组元素<10^9,所以最多有30位,对于每一个二进制位,查找数组中元素在这个位置上有多少1,多少0,假如有 x个1,那么就有nums.size()-x个0
,所以这个位置的个数就是 x*(nums.size()-x),最后累加每一位上的个数即可。
时间复杂度:O(31*n)
空间复杂度:O(1)
class Solution {
public:
int totalHammingDistance(vector<int>& nums) {
int ans=0;
for(int i=30;i>=0;i--)
{
int count=0;
for(auto x: nums)
{
if(x&(1<<i)) count++; //第i位是1的个数
}
ans+=count*(nums.size()-count);
}
return ans;
}
};