477. Total Hamming Distance

solution477.py

# https://leetcode.com/problems/total-hamming-distance/description/

class Solution(object):
    def totalHammingDistance(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        
        i = 0
        total_dist = 0
        while i < 32:
            a = 0
            b = 0
            for num in nums:
                x = num >> i
                x = x & 1
                
                if x == 1:
                    a += 1
                else:
                    b += 1
                    
            total_dist += a*b
            i += 1
                
        return total_dist

nice simple solution