【动态规划】LeetCode算法题：338. 比特位计数

题目

给你一个整数 n ，对于 0 <= i <= n 中的每个 i ，计算其二进制表示中 1 的个数 ，返回一个长度为 n + 1 的数组 ans 作为答案。

思路

从整数$n$到整数$n+1$有两种情况：

1. 不进位：那么整数$n+1$二进制表示中1的个数比整数$n$多一个
2. 进位：先计算（n+1）&1，得到整数$n+1$二进制表示中不受进位影响的数，整数$n+1$二进制表示中1的个数=不受影响的个数+1

阶段划分

每个数当作一个阶段

状态参数

$dp[i]$表示整数$i$的二进制表示中的1的个数

状态转移方程

1. 如果$i$&1==1，表示不进位：$dp[i]=dp[i-1]+1$
2. 否则，表示进位：$tmp=(n+1)$&$1$，$dp[i]=dp[tmp]+1$

代码

def countBits(n: int):
    dp = [0] * (n+1)
    for num in range(1, n+1):
        # 不进位的情况
        if num & 1 == 1:
            dp[num] = dp[num-1] + 1
        # 进位的情况
        else:
            tmp = num & (num - 1)
            dp[num] = dp[tmp] + 1
    return dp