461. Hamming Distance + 477. Total Hamming Distance

▶ 与 Hamming  距离相关的两道题。

▶ 461. 求两个数 x 与 y 的哈夫曼距离。

● 代码，4 ms，对 x 和 y 使用异或，然后求值为 1 的位的个数。

class Solution
{
public:
    int hammingDistance(int x, int y)
    {
        int i,temp;
        for (i = 0, temp = x ^ y; temp > 0; temp &= temp - 1, i++); 
        return i;
    }
};

● 代码，7 ms，转化为 bitset 来逐位比较。

class Solution
{
public:
    int hammingDistance(int x, int y)
    {
        bitset<32> xBS(x), yBS(y);
        int i, count;
        for (i = count = 0; i < 32; i++)
            count += (xBS[i] != yBS[i]);
        return abs(count);
    }
};

 

▶ 477. 求数组 nums 中，所有数字之间两两哈夫曼距离的和。

● 代码，56 ms 。逐位计数计算。把所有数并排放，研究每一位上的 1 有多少个，求其两两异或的结果，如一共有 12 个数，某一位上共 7 个 1，则该位对哈夫曼距离的贡献为 C₇¹C₅¹ = 35 。枚举所有数对进行计算的方法超时了。

class Solution
{
public:
    int totalHammingDistance(vector<int>& nums)
    {
        int i, j, one, n, output; 
        for (i = output = 0, n = nums.size(); i < 32; i++)
        {
            one = 0;
            for (j = 0; j < n; j++)
                one += (nums[j] >> i) & 1;
            output += one * (n - one);
        }
        return output;
    }
};