461. Hamming Distance
本题需要求出两个整数的汉明距离,汉明距离是信息论中的一个概念,表示两个(相同长度)二进制数对应位不同的数量。
本题使用位操作,只需要统计两个变量抑或的结果的二进制中1的个数。
#include <iostream>
using namespace std;
class Solution
{
public:
int hammingDistance(int x, int y)
{
int ret = 0;
int bin = x ^ y;
while (bin != 0) {
ret += bin & 1;
bin >>= 1;
}
return ret;
}
};
191. Number of 1 Bits
191题是上一题的相关题,要求求一个32位整数的汉明重,汉明重即这个数字的二进制中1的个数,代码与461很类似。
#include <iostream>
using namespace std;
class Solution
{
public:
int hammingWeight(uint32_t n)
{
int ret = 0;
while (n != 0) {
ret += n & 1;
n >>= 1;
}
return ret;
}
};
477. Total Hamming Distance
477题求一组数字中所有数字之间的汉明距离之和,有了461题的代码,一个直接的想法是两个求汉明距离再加总,假设有n个数字,需n(n-1)/2次调用hammingDistance,是一个O(n^2)的算法,时间复杂度较高,提交后提示“Time Limit Exceeded”,显然,这是一个Medium难度的题目,必须找出复杂度更低的方法。
该问题求两两之间的汉明距离之和,使用分治策略难以解决,需要考虑汉明距离自身的特点,计算汉明距离的过程其实是先求判断应位之间是否相等,若相等则对结果的贡献是1,否则是0,所以不同位之间完全不相关,可以对这些32位数逐个位来计算,这是这一题的关键。考虑某一位的情况,如果这一位是1,那它与对应位是0的其他数字都会产生1的贡献,反过来这一位是0,则与对应位是1的其他数字都产生1的贡献,也就是说,只需要统计这组数字的这一位是1的数字有几个,是0的数字有几个,两者相乘则是这一位对整个汉明距离的贡献。这是一个O(n)复杂度的答案。
#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
int totalHammingDistance(vector<int>& nums) {
int ret = 0;
int zero_n, one_n;
for (int i = 0; i < sizeof(int) * 8; ++i) {
zero_n = one_n = 0;
for (std::vector<int>::iterator it = nums.begin(); it != nums.end(); ++it) {
one_n += *it & 1;
*it >>= 1;
}
zero_n = nums.size() - one_n;
ret += one_n * zero_n;
}
return ret;
}
};