leetcode 汉明距离总和 Python

汉明距离总和

题目

题目链接：https://leetcode-cn.com/classic/problems/total-hamming-distance/description/

两个整数的汉明距离指的是这两个数字的二进制数对应位不同的数量。

给你一个整数数组 nums，请你计算并返回 nums 中任意两个数之间汉明距离的总和。

示例 1：

输入： nums = [4,14,2]

输出： 6

解释： 在二进制表示中，4 表示为0100 ，14 表示为 1110 ，2表示为0010 。（这样表示是为了体现后四位之间关系） 所以答案为： HammingDistance(4, 14) + HammingDistance(4, 2) + HammingDistance(14, 2) = 2 + 2 + 2 = 6

示例 2：

输入： nums = [4,14,4]

输出： 4

提示：

$$1<=num.length<=10^5$$

$$0<=nums[i]<=10^9$$

解题

思路分析

1. 该问题第一步应当是将全部输入的数据转换为二进制处理，这里可以用到python的bin()，但是应当注意是用[2:]去掉转后后留下的0b
2. 其次应当注意的是将不同位数的二进制数变为相同位数的二进制数（位数少的二进制前方补0）
3. 汉明距离的判断，其实就是各位之间不相同数的数量。例如[0,1,0]那么其汉明距离应该是2(第一位和第二位不同，第二位和第三位不同)
4. 数组的判断，不能采用两两比较的方式去判断，因为这样会大大加大复杂度。
5. 因为二进制只有两种数字，所以其不同数字之和的个数应该是0的个数乘以1的个数，如果全部相同，则必有一个数字的个数为0，那么其相乘结果也必为0。

代码实现

class Solution:
    def totalHammingDistance(self, nums: list[int]) -> int:
        s = 0
        b = list(map(lambda x: bin(x)[2:], nums))
        l_b = list(map(lambda x: len(x), b))
        max_l = max(l_b)
        b = list(map(lambda x: '0' * (max_l - len(x)) + x, b))
        for i in range(max_l):
            temp_0, temp_1 = 0, 0
            for j in b:
                if j[i] == '0':
                    temp_0 += 1
                else:
                    temp_1 += 1
            s += temp_0 * temp_1
        return s

运行结果