477.汉明距离总和.go

/*
 * @lc app=leetcode.cn id=477 lang=golang
 *
 * [477] 汉明距离总和
 *
 * https://leetcode-cn.com/problems/total-hamming-distance/description/
 *
 * algorithms
 * Medium (40.22%)
 * Likes:    37
 * Dislikes: 0
 * Total Accepted:    1.5K
 * Total Submissions: 3.8K
 * Testcase Example:  '[4,14,2]'
 *
 * 两个整数的 汉明距离 指的是这两个数字的二进制数对应位不同的数量。
 *
 * 计算一个数组中，任意两个数之间汉明距离的总和。
 *
 * 示例:
 *
 *
 * 输入: 4, 14, 2
 *
 * 输出: 6
 *
 * 解释: 在二进制表示中，4表示为0100，14表示为1110，2表示为0010。（这样表示是为了体现后四位之间关系）
 * 所以答案为：
 * HammingDistance(4, 14) + HammingDistance(4, 2) + HammingDistance(14, 2) = 2
 * + 2 + 2 = 6.
 *
 *
 * 注意:
 *
 *
 * 数组中元素的范围为从 0到 10^9。
 * 数组的长度不超过 10^4。
 *
 *
 */
func totalHammingDistance(nums []int) int {
	cnt := 0
	mask := 0x01
	for i := 0; i < 32; i++ {
		zeroCount, oneCount := 0, 0
		m := mask << uint(i)
		for _, num := range nums {
			if m&num == m {
				oneCount++
			} else {
				zeroCount++
			}
		}
		cnt += oneCount * zeroCount
	}
	return cnt
}