LeetCode 338 Counting Bits

来源:互联网 发布:网络优化公司智搜宝 编辑:程序博客网 时间:2024/06/18 12:05

题意

给出一个num,输出0到num整数二进制形式中包含1的个数

要求

时间复杂度比简单方法的O(n*sizeof(integer))好
空间复杂度O(n)

题解

0:000
1:001
2:010
3:011
4:100
5:101
递推公式:
res[1]=res[0]+1;
res[2]=res[0]+1;
res[3]=res[1]+1;
res[4]=res[0]+1;
res[5]=res[1]+1;
原理:每次去掉i的二进制形式中最高位的1,取去掉后的数中1的个数,加1就是原数1的个数。

class Solution {public:    vector<int> countBits(int num) {        vector<int>res;        res.resize(num+1);        res[0] = 0;        int count = 0,mod;        for(int i = 1;i <= num;++ i)        {            if(count == 0)            {                count = i;                mod = i;            }            count --;            res[i] = res[i % mod] +1;        }        return res;    }};
0 0
原创粉丝点击