Leetcode :477. 汉明距离总和(位运算）

477. 汉明距离总和

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

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

示例:

输入: 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。

这道题我刚开始用的暴力法，两层循环，毫无疑问错误
解析中先将全部数字每一位的1累加存在一起，之后遍历存1的数组，
相加就OK了
用了两个O(n)的循环，没有超时

class Solution {
public:
    int totalHammingDistance(vector<int>& nums) {
        if(nums.size()==0)
        return 0;
        int n=nums.size();
        int ans=0;
        vector<int>cnt(32,0);
        for(auto num:nums){
            int i=0;
            while(num){
                cnt[i]+=num&1;
                num>>=1;
                i++;
            }
        }
        for(auto k: cnt){
            ans+=k*(n-k);
        }
    return ans;

    }
};