dp基础之位移问题

问题:求从0到N(包括N)的二进制表示里有多少个1?

动态规划解决,分析:要求N的二进制里有多少个1,可以先求N去掉最后一位(N mod 2)后Y(Y = X>>1)里有多少个1
子问题:f[i]表示数字i的二进制里有多少个1
f[i] = i>>1 + i mod 2
数字i的二进制里有多少个1 = i去掉最后一位里的二进制有多少个1 + i的最后一位(有可能是1或0)

初始状态f[0] = 0,
计算顺序:从f[0]...f[N]

时间复杂度达到O(n)

代码及注释如下:

def Number_Of_One(N):
    #f[i]表示数字i的二进制里有多少个1
    f = [0 for i in range(N+1)]
    for i in range(1,N+1):
        f[i] = f[i>>1] + i%2
    return f
N = 5
print(Number_Of_One(N))

结果是:[0, 1, 1, 2, 1, 2]

猜你喜欢

转载自blog.csdn.net/lerry13579/article/details/84030379
今日推荐