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]

在這裡插入圖片描述

解題思路

偶數時,比如二進位制 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; } };