原题地址:https://leetcode-cn.com/problems/total-hamming-distance/submissions/
题目描述:
两个整数的 汉明距离 指的是这两个数字的二进制数对应位不同的数量。
计算一个数组中,任意两个数之间汉明距离的总和。
示例:
输入: 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:
def totalHammingDistance(self, nums: List[int]) -> int:
res = 0
for i in range(30):
tmp = 0
for num in nums:
if num >> i & 1:
tmp += 1
res += tmp * (len(nums) - tmp)
return res