LeetCode 338 比特位计数(动态规划+位运算)

首先，题干如下：

这一题要求给定数num从0—>num的二进制位的数组并返回该数组。最直观的解法就是直接求每个数的二进制位中1的个数，并返回数组。题解如下：

直观解法：

class Solution {
public:    //直接计算
    vector<int> countBits(int num) {
        vector<int> numOfBit(num+1,0);
        for(int i = 0;i<=num;i++){
            numOfBit[i] = fun(i);
        }
        return numOfBit;
    }
    int fun(int x){  //求x的二进制位有多少个1
        int count = 0;
        while(x){
            x = x&(x-1);
            count++;
        }
        return count;
    }
};

这也是我使用的方法，但是题干希望我们能够将时间复杂度降低到O(n)，显然，还有比上面解法更优的解法。思考了一阵子，仍然没有想出来。看了官方题解之后，果然又被安排的明明白白。法宝就是—>位运算。由于之前并不很关注位运算这种解题思路，所以在此记录一下，该题也非常典型。

第一种方法：动态规划+最高有效位

这里首先需要理解最高有效位，最高有效位也就是一个数最左边为1的那个位为最高有效位。假设要求x的二进制位为1的个数bit[x]，其最高有效位为1的数为y，则bit[y] = 1, 那么有等式：bit[x] = bit[x - y] + 1；
但是使用该等式的前提条件是，我们得不断更新每个数的最高有效位。最高有效位的判定条件如下: y&(y-1) == 0；(我们知道，位运算y&(y-1)可以把y最右边的1变为0，由于最高有效位只有一个1，所以经过该位运算之后，y将变为0)
代码如下：

class Solution {
public: //最高有效位
    vector<int> countBits(int num) {
        vector<int> ans(num+1,0);
        int highBit = 0;   //最高有效位
        for(int i = 1;i<=num;i++){
            if((i&(i-1)) == 0){  //判断是否只有一个二进制位为1
                highBit = i;  //更新最高有效位
            }
            ans[i] = ans[i-highBit] + 1;  //转移方程
        }
        return ans;
    }
};

方法二：动态规划+最低有效位

求bit[x]，将x/2，等价于将x>>1，也就是把x最低位去掉，如果最低位为1,那么bit[x] = bit[x >>1] + 1；如果最低位为0，那么bit[x] = bit[x>>1]；
根据以上转移方程，可以写出如下解法：

class Solution {
public:  //最低有效位
    vector<int> countBits(int num) {
        vector<int> ans(num+1,0);
        for(int i=  1;i<=num;i++){
            ans[i] = ans[i >> 1] + (i&1);   //转移方程
        }
        return ans;
    }
};

方法三：动态规划+最低设置位

根据转移方程:bit[x] = bit[x&(x-1)] + 1;

class Solution {
public: //最低设置位
    vector<int> countBits(int num) {
        vector<int> ans(num+1,0);
        for(int i = 1;i<=num;i++){
            ans[i] = ans[i&(i-1)] + 1;
        }
        return ans;
    }
};

这一题可以说是把位运算运用的十分通透了，整理一下，消化一下！！