338. Counting Bits

题目

338. Counting Bits

解题思路

把自然数和对应的二进制数对应起来：

自然数	二进制	bit
0	0	0
1	1	1
2	10	1
3	11	2
4	100	1
5	101	2
6	110	2
7	111	3
8	1000	1
9	1001	2
10	1010	2
11	1011	3
12	1100	2
13	1101	3
14	1110	3
15	1111	4
16	10000	1

这道题的tag是动态规划，所以用动态规划的思路想一下，比如说思路可能是i+1的结果是建立在i之上，仔细观察表格发现：

• 15(1111)的bit数比7(111)多1(多最高位的那个1，后面的同理)，7(111)的bit数比3(11)多1，3(11)的bit数比1(1)多1

• 15比7大8(2^3)，7比3大4(2^2)，3比1大2(2^1)

（或： 13(1101)的bit数比5(101)多1，5(101)的bit数比1(01)多1， 并且： 13比5大8(2^3)，5比1大4(2^2) )

• 1比0大2^0，2~3比0~1大2^1，4~7比0~3大2^2，8~15比0~7大2^3

从以上规律可知，数字i的bit数比数字(i-2^k)(k≥0)多最高位的一个1，并且当 i+1 == 2 ^ (k+1)时，k++；

具体代码

class Solution {
public:

    vector<int> countBits(int num) {
        vector<int> nums;
        nums.push_back(0);
        int bit = 1;  // 2^0
        for (int i = 1; i <= num; i++) {
            // 数字i的bit数比数字(i-bit)多最高位的一个1
            nums.push_back(nums[i - bit] + 1);    
            if (i + 1 == bit * 2)
                bit *= 2;
        }

        return nums;
    }
};

测试代码

#include <iostream>
#include <vector>
using namespace std;

vector<int> countBits(int num) {
    vector<int> nums;
    nums.push_back(0);
    int bit = 1;
    for (int i = 1; i <= num; i++) {
        nums.push_back(nums[i - bit] + 1);    
        if (i + 1 == bit * 2)
            bit *= 2;
    }

    return nums;
}

void test() {
    int num = 16;
    vector<int> nums = countBits(num);
    for (int i = 0; i <= num; i++)
        cout << nums[i] << " ";
    cout << endl;
}

int main() {
    test();
}