1. 程式人生 > >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.

注意:

  1. 陣列中元素的範圍為從 0到 10^9
  2. 陣列的長度不超過 10^4

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