aboutsummaryrefslogtreecommitdiff
path: root/total_hamming_distance.py
diff options
context:
space:
mode:
Diffstat (limited to 'total_hamming_distance.py')
-rw-r--r--total_hamming_distance.py27
1 files changed, 27 insertions, 0 deletions
diff --git a/total_hamming_distance.py b/total_hamming_distance.py
new file mode 100644
index 0000000..3098372
--- /dev/null
+++ b/total_hamming_distance.py
@@ -0,0 +1,27 @@
+# 477. Total Hamming Distance
+
+"""
+the hamming distance between two integers is the number of positions at which
+the corresponding bits are different. given an integer array 'nums', return
+the sum of hamming distances between all the pairs of the integers in 'nums'.
+"""
+
+
+class Solution(object):
+ def totalHammingDistance(self, nums):
+ """
+ :type nums: List[int]
+ :rtype: int
+ """
+ bits = [[0, 0] for _ in range(32)]
+ for i in nums:
+ for j in bits:
+ j[i % 2] += 1
+ i /= 2
+ return sum(x * y for x, y in bits)
+
+
+if __name__ == "__main__":
+ obj = Solution()
+ print(obj.totalHammingDistance(nums=[4, 14, 2]))
+ print(obj.totalHammingDistance(nums=[4, 14, 4]))