这几天零星做到了一些跟正整数位有关的操作的题目,就常见的就是异或,都需要 O(n) 的时间复杂度。所以把这些题目总结一下。
477. Total Hamming Distance
数字二进制的Hamming距离为,两个数字异或之后数中1的个数。所以让求出,一个数组中,所有数两两之间Hamming距离的和。
421. Maximum XOR of Two Numbers in an Array
一个数组中,找出两个数异或得到的数最大的结果。对于这种问题,因为都涉及到数组中所有数的两两运算,按道理来说的话应该是
O(n2)
的时间复杂度,但是题目中要求
O(n)
的时间复杂度,所以就需要换一个角度去想这个问题。因为是按位操作的,所以可以考虑一位一位的来,这样一个int有32位,所以最后复杂度应该是
32×O(n)
即还是
O(n)
的时间复杂度。
对于477题,所有点之间的Hamming距离的和按位来看的话,就是只看其中的某一位,只有不一样的才会贡献Hamming距离,所以找出那一位中1的个数
m
和0的个数
代码如下:
int totalHammingDistance(vector<int>& nums) {
int sum = 0;
for(int i = 0; i<32;i++){
int mark = 1<<i;
int count = 0;
for(auto j:nums)
if(j&mark)
count++;
sum += count*(nums.size()-count);
}
return sum;
}
对于421题,也可以一位一位来,从最高位开始依次往后,假设前n-1位异或出来的最大值res已经找出来了,那么看前第n位:判断(res<<1)|1
是否能够满足凑成,如果满足题意,这第n为1,否则为0。
代码如下:
int findMaximumXOR(vector<int>& nums) {
int res = 0;
for(int i = 31;i>=0;i--){
res = res<<1;
int mark = 0;
//这里要用到hashmap,因为只有hashmap的查找复杂度才为O(1)
unordered_map<int,int> tmp;
for(auto j:nums){
tmp[j>>i]++;
if(tmp.find((j>>i)^1^res) != tmp.end()){
mark = 1;
break;
}
}
res = res | mark;
}
return res;
}
这道题其实标准做法应该是Trie数(字典树)的做法,稍后更新