1046. 二进制表示中质数个计算置位

1046. 二进制表示中质数个计算置位

中文English

给定两个整数 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。

输入测试数据 (每行一个参数)如何理解测试数据？

class Solution:
    """
    @param L: an integer
    @param R: an integer
    @return: the count of numbers in the range [L, R] having a prime number of set bits in their binary representation
    """
    '''
    大致思路：
    1.给出一个方法，取出R的二进制最长多少位，然后根据这个长度，来判断最长1的个数，给出一个方法，在这长度里面取出所有符合质数的列表集合，返回。
    2.初始化count = 0，在L和R之间循环，取出对应的str(二进制数[2:]),然后得到1的个数，判断如果为质数的话，则count += 1，最终返回count即可。
    '''
    def countPrimeSetBits(self,L,R):
        all_prime_dic = self.getprimedic(R)
        count = 0
        for i in range(L,R+1):
            bin_count = self.getonecount(i)
            if bin_count in all_prime_dic:
                count += 1
        return count

    ##质数的列表集合
    def getprimedic(self,R):
        dic = []
        res = []
        l = len(str(bin(R+1))[2:])
        for i in range(2,l):
            if i not in dic:
                ##此时得到的是不是质数的集合
                for j in range(i*2,l,i):
                    dic.append(j)
        
        for z in range(2,l):
            if z not in dic:
                res.append(z)
        return res
        
    def getonecount(self,num):
        return str(bin(num))[2:].count('1')