力扣477(java)-汉明距离总和(中等)
题目:
两个整数的 汉明距离 指的是这两个数字的二进制数对应位不同的数量。
给你一个整数数组 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 <= nums.length <= 104
0 <= nums[i] <= 109
给定输入的对应答案符合 32-bit 整数范围
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/total-hamming-distance
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路:
逐位统计
统计出每个位置上1和0的个数,例如长度为 n 的数组 nums 的所有元素二进制的第 i 位共有 s1个 1, n-s1个 0,对于每个1来说,与0的汉明距离都是1,与1的汉明距离为0,则这些元素在二进制的第 i 位上的汉明距离之和为 s1*(n−s1)。
代码:
1 class Solution { 2 public int totalHammingDistance(int[] nums) { 3 int res = 0, n = nums.length; 4 for(int i = 0; i < 32; i++){ 5 int s1 = 0; 6 for(int v: nums){ 7 s1 += (v >> i) & 1; 8 } 9 res += s1 * (n -s1); 10 } 11 return res; 12 13 } 14 }