LeetCode 762. 二进制表示中质数个计算置位(C++、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 不是质数)

注意:

L, R 是 L <= R 且在 [1, 10^6] 中的整数。

R - L 的最大值为 10000。

C++

class Solution {
public:
    bool prime(int n)
    {
        if(n<=1)
        {
            return false;
        }
        else
        {
            int right=(int)sqrt(n);
            for(int i=2;i<=right;i++)
            {
                if(0==n%i)
                {
                    return false;
                }
            }
            return true;
        }
    }

    int countPrimeSetBits(int L, int R) 
    {
        int res=0;
        for(int i=L;i<=R;i++)
        {
            int num=i;
            int count=0;
            while(num)
            {
                if(1==num%2)
                {
                    count++;
                }
                num/=2;
            }
            if(prime(count))
            {
                res++;
            }
        }
        return res;
    }
};

python

class Solution:
    def prime(self,n):
        if n<=1:
            return False
        else:
            right=int(n**0.5)
            for i in range(2,right+1):
                if 0==n%i:
                    return False
            return True

    def countPrimeSetBits(self, L: int, R: int) -> int:
        res=0
        for i in range(L,R+1):
            count=0
            num=i
            while num:
                if 1==num%2:
                    count+=1
                num//=2
            if self.prime(count):
                res+=1
        return res
        

猜你喜欢

转载自blog.csdn.net/qq_27060423/article/details/89188029