Leetcode: 477. Total Hamming Distance
主要是为了熟悉位操作来了。这题是基本题,考察hamming距离。定义两个数字间的hamming距离为两个数字间相应位不同的bit数。
背景
按位与&、按位或|、异或^、取反~、左移<<、右移>>、无符号右移>>>。这有一篇很好的博文,主要还把一般的用法都总结了一下。
代码
一个简单的想法是,先把两个数字做异或运算,然后算结果有几个1就可以了。
一个更加一般化的想法是,对于一个数字集,两两算出他们的hamming距离最后相加就可以了。但是这种方法明显冗余了,有没有办法更加节省时间呢?
尝试把set中的数字看作一个整体,把最低位分为0,1两部分,已知内部集合两个数字最低位的hamming距离肯定是0,所以主要看两个集合相交的可能数。然后右移一位,接着算,直到算完所有位。这就把按位异或问题转为了按位分类。在对比的次数上节省了不少。
class Solution {
public:
int totalHammingDistance(vector<int>& nums) {
int size = nums.size();
if(size < 2) return 0;
int ans = 0;
int *zeroOne = new int[2];
while(true)
{
int zeroCount = 0;
zeroOne[0] = 0;
zeroOne[1] = 0;
for(int i = 0; i < nums.size(); i++)
{
if(nums[i] == 0) zeroCount++;
zeroOne[nums[i] % 2]++;
nums[i] = nums[i] >> 1;
}
ans += zeroOne[0] * zeroOne[1];
if(zeroCount == nums.size()) return ans;
}
}
};