1. 问题描述:
两个整数的汉明距离指的是这两个数字的二进制数对应位不同的数量。计算一个数组中,任意两个数之间汉明距离的总和。
示例:
输入: 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
。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/total-hamming-distance
2. 思路分析:
分析题目可以知道最容易想到的是使用两层循环枚举当前数字与其余的两个数字,计算他们二进制不同的位数,但是由于数组的最大长度为10 ^ 4所以两层循环应该会超时的,所以我们不能够枚举任意两个数字。这里可以枚举每一位,将所有数字当前这一位分为两组,一组是所有数字当前这一位为0的数目x,另外一组是所有数字当前这一位为1的数目y,所以这一位不同二进制位的数目为x * y,int类型最大是31位,所以枚举31位即可。
3. 代码如下:
from typing import List
class Solution:
def totalHammingDistance(self, nums: List[int]) -> int:
res = 0
# 枚举每一位
for i in range(31):
# 枚举每一个数字, x表示当前第i位0的数目, y表示当前第i位1的数目
x, y = 0, 0
for j in range(len(nums)):
if nums[j] >> i & 1:
y += 1
else:
x += 1
res += x * y
return res