1. 程式人生 >實用技巧 >其它 >LeetCode 338. 位元位計數(C++)動態規劃

LeetCode 338. 位元位計數(C++)動態規劃


技術標籤:LeetCode刷題leetcode動態規劃演算法c++資料結構

給定一個非負整數 num。對於 0 ≤ i ≤ num 範圍中的每個數字 i ,計算其二進位制數中的 1 的數目並將它們作為陣列返回。

示例 1:

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

示例 2:

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

LeetCode 338. 位元位計數(C++)動態規劃

解題思路

偶數時,比如二進位制 1010 其1的個數和它除以2的是一樣的,即和 101 帶1個數一致
奇數時,加一即可,可以得出

res[i] = res[i >> 1] + (i & 1)

class Solution {
public:
    vector<int> countBits(int num) {
        vector<int> res(num + 1);//初始化時 每個位都是0
        for (int i = 1; i <= num; ++i)
        //>>是右bai移運算子。假設x=5,那麼x的二進位制為0101,x>>1表示x右移1位,即把最右邊一位的1刪掉,變為010,此時x=2;
            res[i] = res[i >> 1] + (i & 1);//  res[i >> 1]  偶數時,比如二進位制 1010 其1的個數和它除以2的是一樣的
            //奇數時,加一即可,可以得出
        return res;
    }
};