LeetCode -- 338. 比特位计数

链接：比特位计数

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

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

要求：

• 给出时间复杂度为O(n*sizeof(integer))的解答非常容易。但你可以在线性时间O(n)内用一趟扫描做到吗？
• 要求算法的空间复杂度为O(n)。
• 你能进一步完善解法吗？要求在C++或任何其他语言中不使用任何内置函数（如 C++ 中的 __builtin_popcount）来执行此操作。

思路：一开始想到的是使用for循环遍历0~num的数i，并在循环内使用/2，%2得出i的数字1的数目，或者同理使用递归，但如此以来时间复杂度绝对超过了O(n)。这时可以使用一个奇淫技巧：位运算。思路是动态规划。

代码：

class Solution {
public:
    vector<int> countBits(int num) {
    	vector<int>one(num+1);
		for(int i = 0; i <= num ; i++)
		{
			one[i] = one[i >> 1] + (i & 1); //one[i] = [i >> 1所包含的1的个数] + [i的最低位是否为1] 
		}
		return one;
    }
};

位运算这一工具十分强大，若只求一个数的1的位数和，时间复杂度甚至就为O(1).

---

在此为位运算做一些小笔记：

位运算基础

• &（与）
  两个相应的二进制位都为1，则该位的结果值为1，否则为0
• |（或）
  两个相应的二进制位中只要有一个为1，该位的结果值为1
• ^（异或）
  若参加运算的两个二进制位值相同则为0，否则为1
• ~ （非 / 取反）
  一元运算符，用来对一个二进制数按位取反，即将0变1，将1
• <<（将二进制位全部左移N位，右补0）
• >> (将二进制位进行右移操作)
  移到右端的低位被舍弃，对于无符号数， 高位补0
• >>>运算符将用0填充高位；>>运算符用符号位填充高位，没有<<<运算符
• 对于int型，1<<35 与 1<<3是相同的

位运算部分巧妙用法

• 判断奇偶数：x & 1 = 1 为奇数 ，= 0为偶数
• 交换两个整数变量的值

 public static void main(String[] args) {
        int a = 5;
        int b = 3;
        a = a ^ b;
        b = a ^ b;
        a = a ^ b;
        System.out.println(a);
        System.out.println(b);     
    }
}

• 获取二进制位是1还是0

    public static void main(String[] args) {
    	int x = 4;  //二进制为0100
    	int y = 3;
    	System.out.println(x + "的第" + y + "位的二进制位为：" + ( ((x & (1<<(y-1)))>>(y-1)) == 0 ? "0":"1"));
    } 	  
   }

• 不用判断语句，求整数的绝对值
  原理：将一个整型整数x，带符号右移31位，则结果要么是0，要么是-1。如果是0，则x为正数，为-1则x为负数。
  然后，将x与右移31位后的结果做异或运算，当与x^0是，结果还是x。 当x^-1时，结果为x取反，即x的反码，然后+1，即为x的绝对值。
• 用O(1)时间检测整数n是否是2的幂次
  一位N的二进制表示中只有一个1，所以使用N&(N-1)将唯一的一个1消去。
  如果N是2的幂次，那么N&(N-1)得到结果为0，即可判断。
• 计算在一个 32 位的整数的二进制表示中有多少个 1
  由 x & (x-1) 消去x最后一位知。循环使用x & (x-1)消去最后一位1，计算总共消去了多少次即可。