Leetcode: 477. Total Hamming Distance

最新推荐文章于 2023-08-09 20:49:23 发布

小森林冬春

最新推荐文章于 2023-08-09 20:49:23 发布

阅读量272

点赞数

分类专栏： LeetCode 文章标签： leetcode 位运算

本文链接：https://blog.csdn.net/qq_16318319/article/details/73734766

版权

LeetCode 专栏收录该内容

20 篇文章 0 订阅

订阅专栏

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;
        }
    }
};