LeetCode：338（Python）—— 比特位计数（简单）

比特位计数

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

输入：n = 2
输出：[0,1,1]

输入：n = 5
输出：[0,1,1,2,1,2]

方法一：count() 方法

思路：首先把整型转为二进制，然后用 count() 统计字符串中 1 的个数，依次循环即可。

# count方法
# 首先把整型转为二进制，然后用count()统计字符串中 1 的个数，依次循环即可。
class Solution:
    def countBits(self, n: int) -> List[int]:
        ans = []
        for i in range(n + 1):
            ans.append(bin(i).count('1'))
        return ans

方法二：Brian Kernighan 算法

思路：该算法把二进制最后一个 1 变为 0，然后依次循环添加即可。

# Brian Kernighan算法
# 该算法把二进制最后一个 1 变为 0，然后依次循环添加即可。
class Solution:
    def countBits(self, n: int) -> List[int]:
        def count_ones(x: int ) -> int:
            ones = 0
            while x > 0:
                x &= (x - 1)
                ones += 1
            return ones
        return [count_ones(i) for i in range(n + 1)]

方法三：最高有效位

思路：定义整数 high_bit 来统计最高有效位，两者差值 +1 即为个数，依次循环添加即可。

# 最高有效位
# 定义整数high_bit来统计最高有效位，两者差值 +1 即为个数，依次循环添加即可。
class Solution:
    def countBits(self, n: int) -> List[int]:
        bits = [0]
        high_bit = 0
        for i in range(1, n + 1):
            if i & (i - 1) == 0:
                high_bit = i
            bits.append(bits[i - high_bit] + 1)
        return bits

方法四：最低有效位

思路：每次将二进制右移一位，并加上除 2 余数，依次循环添加即可。

# 最低有效位
# 每次将二进制右移一位，并加上除 2 余数，依次循环添加即可。
class Solution:
    def countBits(self, n: int) -> List[int]:
        bits = [0]
        for i in range(1, n + 1):
            bits.append(bits[i >> 1] + (i & 1))
        return bits

方法五：最低设置位

思路：直接取最低位置从 1 变成 0 以后的数，依次循环添加即可。

# 最低设置位
# 直接取最低位置从 1 变成 0 以后的数，依次循环添加即可。
class Solution:
    def countBits(self, n: int) -> List[int]:
        bits = [0]
        for i in range(1, n + 1):
            bits.append(bits[i & (i - 1)] + 1)
        return bits

总结

好家伙，官解就是大气，三种动归直接给我当头一棒，嘲讽我的无能！