- 比特位计数
给定一个非负整数 num。对于 0 ≤ i ≤ num 范围中的每个数字 i ,计算其二进制数中的 1 的数目并将它们作为数组返回。
示例 1:
输入: 2
输出: [0,1,1]
示例 2:
输入: 5
输出: [0,1,1,2,1,2]
思路解析:
1·简单的思路就是一个数一个数地算,就是逐位判断是否为1,之后累加。用移位也好用取余数也好,都是这么个思路。代码就不赘述了。复杂度O(n*sizeof(integer))
2·用一种O(n)的方式,一边扫描。其实不难发现二进制数字的规律。0中“1”的个数是0,作为特例。往后每一个数字,其实都符合一个规律。如果是奇数,那“1”的个数就是上一个数(也就是偶数)中“1”的个数+1;如果是偶数,那其“1”的个数就是其左移一位,即x/2那个数中“1”的个数。
这应该也算是一种简单的动态规划,建表查表嘛。代码如下:
// leecode338.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
//
#include <iostream>
#include<vector>
using namespace std;
vector<int> countBits(int num)
{
vector<int> array;
array.push_back(0);
for (int i = 1; i <= num; i++)
{
if (i % 2 == 1)
{
array.push_back(array[i - 1] + 1);
}
else
{
array.push_back(array[i >> 1]);
}
}
return array;
}
int main()
{
int num;
while(cin >> num)
{
vector<int> array = countBits(num);
for (auto vec : array)
{
cout << vec << endl;
}
cout << "********************" << endl;
}
return 0;
}
提交结果如下(只需要提交countBits函数部分),看起来应该是蛮快的。
但是,有大神看见了的话,还希望请教比这个还快的那一部分,是哪里快了。
哪怕我用三目运算符,好像也没快啊!
vector<int> countBits(int num)
{
vector<int> array;
array.push_back(0);
for (int i = 1; i <= num; i++)
(i % 2 == 1) ? array.push_back(array[i - 1] + 1) : array.push_back(array[i >> 1]);
return array;
}
而且可读性也不如直接写if…else