[Leetcode] 第338题 比特位计数

 

一、题目描述

给定一个非负整数 num。对于 0 ≤ i ≤ num 范围中的每个数字 i ，计算其二进制数中的 1 的数目并将它们作为数组返回。

示例 1:

输入: 2
输出: [0,1,1]

示例 2:

输入: 5
输出: [0,1,1,2,1,2]

二、题目分析

1）发现规律：2的幂1的数目一定是1个，用last_2记录最近出现的2的幂
2）dp[i]=1+dp[i-last_2]，比如dp[7]=1+dp[3]

三、代码实现

class Solution {
public:
    vector<int> countBits(int num) {
        vector<int>res(num + 1);
        res[0] = 0;
        if (num < 1)return res;
        res[1] = 1;
        int last_2 = 1;
        for (int i = 2; i <= num; ++i) {
            if (i%last_2 == 0 && i / last_2 == 2) {
                res[i] = 1;
                last_2 = i;
            }
            else
                res[i] = 1 + res[i - last_2];
        }
        return res;
    }
};