leetcode 762. 二进制表示中质数个计算置位(python)

给定两个整数 L 和 R ，找到闭区间 [L, R] 范围内，计算置位位数为质数的整数个数。

（注意，计算置位代表二进制表示中1的个数。例如 21 的二进制表示 10101 有 3 个计算置位。还有，1 不是质数。）

示例 1:

输入: L = 6, R = 10
输出: 4
解释:
6 -> 110 (2 个计算置位，2 是质数)
7 -> 111 (3 个计算置位，3 是质数)
9 -> 1001 (2 个计算置位，2 是质数)
10-> 1010 (2 个计算置位，2 是质数)

示例 2:

输入: L = 10, R = 15
输出: 5
解释:
10 -> 1010 (2 个计算置位, 2 是质数)
11 -> 1011 (3 个计算置位, 3 是质数)
12 -> 1100 (2 个计算置位, 2 是质数)
13 -> 1101 (3 个计算置位, 3 是质数)
14 -> 1110 (3 个计算置位, 3 是质数)
15 -> 1111 (4 个计算置位, 4 不是质数)

注意:

1. L, R 是 L <= R 且在 [1, 10^6] 中的整数。
2. R - L 的最大值为 10000。

思路一：

因为L,R的数值在[1,1000000]之间，换算一下，其实二进制位数也就是在20位以下。这样的话可以用一个列表把1-20里边所有的质数表示出来。然后在依次判断在区间的每一个数就行。

class Solution(object):
    def countPrimeSetBits(self, L, R):
        """
        :type L: int
        :type R: int
        :rtype: int
        """
        set_bits = [2,3,5,7,11,13,17,19]
        num = 0
        for i in range(L,R+1):
            res = bin(i).count('1')
            if res in set_bits:
                num += 1
        return num

思路二：和思路其实差不多，唯一的亮点是用0,1列表代替了质数列表。

class Solution(object):
    def countPrimeSetBits(self, L, R):
        """
        :type L: int
        :type R: int
        :rtype: int
        """
        # L,R 是在 [1,10^6] 中的整数，因此置位的个数最多为 19
        #创建列表 p，0-20 中质数为置 1 ，非质数位置 0

        p = (0, 0, 1, 1, 0, 1, 0, 1, 0, 0, 0, 1, 0, 1, 0, 0, 0, 1, 0, 1)
        re = 0
        for n in range(L, R + 1):
            re += p[bin(n).count('1')] 
        return re