leetcode477. 汉明距离总和(位运算优化)

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

计算一个数组中，任意两个数之间汉明距离的总和。

示例:

输入: 4, 14, 2

输出: 6

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

注意:

1. 数组中元素的范围为从 0到 10^9。
2. 数组的长度不超过 10^4。

思路：
数据范围为10的四次方，双for去查找每一位的不同距离，将会超时
，这时我们需要优化，对于整体的每个数中的每一位算汉明距离总和，我们就达到减少遍历次数的目的，汉明距离即每一位不同的个数，长度为 n 的数组nums 的所有元素二进制的第 i 位共有 c个 1，n-c 个 0，则些元素在二进制的第 i位上的汉明距离之和为 c*(n-c)

代码1：
（勉强不超时，使用了bitcount（）方法）

class Solution {
        public int totalHammingDistance(int[] nums) {
        int res = 0;
        for(int i = 0; i < nums.length - 1; i++){
            for(int j = i+1;j < nums.length; j++){
                res += getHamm(nums[i],nums[j]);
            }
        }
        return res;
    }
    private int getHamm(int x,int y){
        return Integer.bitCount(x^y);
    }

}

代码2：（优化后）

class Solution {

    // 统计第i位上的1和0的个数，相乘就是该位上产生的汉明距离，并相加,i的位数不超过30位
    public int totalHammingDistance(int[] nums){
        int ans=0;
        for(int i=0;i<30;i++){
            int c=0,l=0;
            for(int j=0;j<nums.length;j++){
                if(nums[j]>>i==1){
                    l++;
                }else{
                    c++;
                }
            }
            ans+=c*l;
        }


        return ans;
    }

}