leetcode 477 汉明距离综合

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.

思路： 汉明距离想要求的是两个数字在2进制数表示下不同的位数，那对于多个数的某一位而言，其实就是讨论有多少个0 or 1，以及他们能够构成多少种不同的组合，因此思路就比较直接了，对位数进行遍历，子循环中遍历每个数的该位置，计算组合数。

假设某一位中有a个0，b个1，不同的组合就是 a * b

class Solution:
    def totalHammingDistance(self, nums: List[int]) -> int:
        n = len(nums)
        res = 0
        for i in range(30):
            a = 0
            for k in nums:
                a += (k >> i) & 1
            res += a * (n - a)
        return res