两个整数的 汉明距离 指的是这两个数字的二进制数对应位不同的数量。
计算一个数组中,任意两个数之间汉明距离的总和。
示例:

  1. 输入: 4, 14, 2
  2. 输出: 6
  3. 解释: 在二进制表示中,4表示为010014表示为11102表示为0010。(这样表示是为了体现后四位之间关系)
  4. 所以答案为:
  5. HammingDistance(4, 14) + HammingDistance(4, 2) + HammingDistance(14, 2) = 2 + 2 + 2 = 6.

注意:

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

暴力超时法

class Solution {
public:
    int totalHammingDistance(vector<int>& nums) {
        int res = 0;
        for(int i = 0; i < nums.size();i++){
            for(int j = i; j < nums.size(); j++){
                res += hammingDistance(nums[i], nums[j]);
            }
        }
        return res;
    }
    int hammingDistance(int x, int y) {
        int count = 0;
        while(!(x== 0 && y == 0)){
            if(x%2 != y%2){
                count++;
            }
            // cout<<x<<" "<<y<<endl;
            x = x /2;
            y = y /2;
        }
        return count;
    }
};

按位计算

class Solution {
public:
    int totalHammingDistance(vector<int>& nums) {
        int res = 0;
        int len = nums.size();
        if (len == 0)
            return 0;
        for (int i = 0; i < 32; ++i)
        {
            int pos = 0;
            int neg = 0;
            for (int j = 0; j < len; ++j)
            {
                if (nums[j] & 1)
                    ++pos;
                else 
                    ++neg;
                nums[j] >>= 1;
            }
            res += (neg * pos);
        }
        return res;
    }
};