两个整数的 汉明距离 指的是这两个数字的二进制数对应位不同的数量。计算一个数组中,任意两个数之间汉明距离的总和。
示例:
输入: 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。
解析:
对于数组 nums 中的某个元素val,若其二进制的第 i 位为 1,我们只需统计 nums 中有多少元素的第 i 位为 0,即计算出了val 与其他元素在第 i 位上的汉明距离之和。
具体地,若长度为 n 的数组 nums 的所有元素二进制的第 i 位共有 c 个 1,n−c 个 0,则些元素在二进制的第 i 位上的汉明距离之和为c*(n-c)
我们可以从二进制的最低位到最高位,逐位统计汉明距离。将每一位上得到的汉明距离累加即为答案。
代码如下:
class Solution {
public:
int totalHammingDistance(vector<int>& nums) {
int sum=0,len=nums.size();
for(int i=0;i<30;i++)//因为10^9<2^30
{
int t=0;
for(auto val:nums) t+=(val>>i)&1;
sum+=t*(len-t);
}
return sum;
}
};
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/total-hamming-distance
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
如有问题,欢迎在评论区提问