【Leetcode 338】 Counting Bits

---恢复内容开始---

问题描述:给出一个非负整数num,对[0, num]范围内每个数都计算它的二进制表示中1的个数

Example:
For num = 5 you should return [0,1,1,2,1,2]

思路:这种题适合归纳法,找出规律然后用编程语言描述,令i从0开始,设f(i)为i对应二进制表示中1的个数

也就是每次有一个大小为2^n的序列,把他们每个数都加1,然后插入到结尾就构成了大小为2^(n+1)个序列。

直到2^(n+1)比num大,取前num个数组成序列返回即可。

解法

刚开始试着用链表list来动态增加(splice),把原来的list(l0)每个值都加1后构成新的list(l1),然后把l1插入到l0末尾。

这样l0大小从1到2到4到8……一直到log2(len),再把剩下的len - log2(len)个元素构成新的list(l1)插入到末尾。

class Solution {
public:
	vector<int> countBits(int num) {
		int len = num + 1;
		list<int> l0 = { 0 };
		int n = 1;
		bool b = true;
		while (n < len) {
		    if (len - n < n) {
		        int newCnts = len - n;
		        auto it = l0.cbegin();
		        for (int i = 0; i < newCnts; ++i) {
		            l0.push_back(1 + *it++);
		        }
		        break;
		    } else {
		       	list<int> l1;
			    auto it = l0.cbegin();
			    while (it != l0.cend())
				l1.push_back(1 + *it++);
			    l0.splice(l0.end(), l1);
			    n *= 2;
		    }
		}
		vector<int> res(len);
		auto it1 = l0.cbegin();
		auto it2 = res.begin();
		while (it2 != res.end())
			*it2++ = *it1++;
		return res;
	}
};  // 15/15 96ms 20.09%

然后发现代码又长,效率不怎样,这里splice比起在vector中计算元素的位置没有什么优势,问题在于每次push_back都要新申请一个元素的空间,最后还要重新转换成vector(PS:leetcode给出的C++函数签名的返回值是vector),这种靠链接list的小伎俩还不如老老实实用vector。(好吧我承认是被list的排序算法影响到了……)

于是用vector写了下,代码很短而且效率还可以

class Solution {
public:
	vector<int> countBits(int num) {
	    const int len = num + 1;
	    vector<int> res(len);

	    int nb = 1;
	    int ne = nb << 1;
	    res[0] = 0;
	    while (ne < len) {
	        for (int i = nb; i < ne; ++i)
	            res[i] = res[i - nb] + 1;
	        nb = nb << 1;
	        ne = nb << 1;
	    }
	    for (int i = nb; i < len; ++i)
	        res[i] = res[i - nb] + 1;

	    return res;
	}
};  // 15/15 88ms 53.16%

试着改进了下,就是对于每个新的序列,比如[4, 8)对应{1, 2, 2, 3},[1, 4)对应{0, 1, 1, 2}

[4, 6)区域和[2, 4)区域的值是一样的,而[6, 8)区域的值是[2, 4)区域对应值加1。这样每次遍历[nb + (ne-nb)/2, ne)再运算两次即可,用temp保留遍历的值,temp和temp+1赋给新的位置。这样省去了一次+1操作。

但是问题是这样增加了一次赋值操作,测试运行时间是一样的,暂时就此打住吧。

改进的话可能有2方面:1、发现新的规律,感觉有点深入数学层次了;2、不要每次都计算等号左右的迭代器位置,而是用2个迭代器一直移动,有空再看吧。

时间: 2024-10-23 12:13:29

【Leetcode 338】 Counting Bits的相关文章

【LeetCode 229】Majority Element II

Given an integer array of size n, find all elements that appear more than ⌊ n/3 ⌋ times. The algorithm should run in linear time and in O(1) space. 思路: [LeetCode 169]Majority Element 的拓展,这回要求的是出现次数超过三分之一次的数字咯,动动我们的大脑思考下,这样的数最多会存在几个呢,当然是2个嘛.因此,接着上一题的方

【LeetCode OJ】Sum Root to Leaf Numbers

? 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 # Definition for a  binary tree node # class TreeNode: #     def __init__(self, x): #         self.val = x #         self.left = No

【LeetCode OJ】Longest Consecutive Sequence

Problem Link: http://oj.leetcode.com/problems/longest-consecutive-sequence/ This problem is a classical problem where we can reduce the running time by the help of hash table. By given a list of numbers, we can find the longest consecutive sequence b

【LeetCode OJ】Word Ladder I

Problem Link: http://oj.leetcode.com/problems/word-ladder/ Two typical techniques are inspected in this problem: Hash Table. One hash set is the words dictionary where we can check if a word is in the dictionary in O(1) time. The other hash set is us

【leetcode系列】String to Integer (atoi)

这个我就直接上代码了,最开始把"abc123"也算作合法的了,后来查了一下atoi的定义,把这种去掉了. public class Solution { public static int atoi(String inStr) { long result = 0L; /* * 网上查了一下,atoi函数的定义是如果第一个非空格字符存在,是数字或者正负号则开始做类型转换, * 之后检测到非数字(包括结束符\0)字符时停止转换,返回整型数.否则,返回零.可能的输入情况有: 1.空字符串 *

【leetcode系列】Valid Parentheses

很经典的问题,使用栈来解决,我这里自己实现了一个栈,当然也可以直接用java自带的Stack类. 自己实现的栈代码: import java.util.LinkedList; class StackOne { LinkedList<Object> data; int top; int maxSize; StackOne(int size) { // TODO Auto-generated constructor stub top = -1; maxSize = 100; data = new

【leetcode系列】3Sum

这个题我最开始的思路是:先一个数定下来,然后在除这个数之外的集合里面找另外两个数,最后计算和.如此反复,对于N个数,需要进行N-2次循环. 我遇到的问题就是怎么找另外两个数,其实我想过参照Two Sum里面的解法,就是用Hashtable存,键值对的结构是<任意两个数的和,<下标1,下标2>>,但是构造这个Hashtable就需要O(N^2),后面真正解的时候有需要O(N^2). 参考了大牛的解法后,明白了找两个数还是用两个下标同时往中间移动比较好,下面上代码. import ja

【leetcode系列】Two Sum

解法一,我自己想的,有点弱,时间复杂度O(n^2). public class Solution { public int[] twoSum(int[] numbers, int target) { int[] result = new int[2]; for (int i = 0; i < numbers.length; i++) { for (int j = i + 1; j < numbers.length; j++) { if (numbers[i] + numbers[j] == t

【leetcode 简单】第十七题 二进制求和

实现 int sqrt(int x) 函数. 计算并返回 x 的平方根,其中 x 是非负整数. 由于返回类型是整数,结果只保留整数的部分,小数部分将被舍去. 示例 1: 输入: 4 输出: 2 示例 2: 输入: 8 输出: 2 说明: 8 的平方根是 2.82842...,   由于返回类型是整数,小数部分将被舍去. #define PF(w) ((w)*(w)) int mySqrt(int x) { int start = 0; int end = x; double mid = 0; i