LeetCode "477. Total Hamming Distance"

Fun one.. the punch line of this problem is quite common in Bit related problems on HackerRank - visualize it in your mind, and you will find: all bits on the same index among all numbers, will not involve other bits on other indices. So, we simply count number of 0s or 1s, on the same bit index from all numbers - we count it vertically..

class Solution {

public:

    int totalHammingDistance(vector<int>& nums)

    {

        int n = nums.size();

        if(n < ) return ;

        // Pass 1: count num of 1s O(31n) -> O(n)

        vector<unsigned> cnt();

        for(auto v : nums)

        for(int i = ; i < ; i ++)

            cnt[i] += (v & ( << i)) ?  : ;

        // Pass 2: count

        int ttl = ;

        for(int i = ; i < ; i ++)

            ttl += cnt[i] * (n - cnt[i]);

        return ttl;

    }

};

巴特西

LeetCode "477. Total Hamming Distance"

最新文章

热门文章