题目描述
给定两个整数 L 和 R ,找到闭区间 [L, R] 范围内,计算置位位数为质数的整数个数。注意,计算置位代表二进制表示中1的个数。例如 21 的二进制表示 10101 有 3 个计算置位。还有,1 不是质数。)
示例
输入: L = 6, R = 10
输出: 4
解释:
6 -> 110 (2 个计算置位,2 是质数)
7 -> 111 (3 个计算置位,3 是质数)
9 -> 1001 (2 个计算置位,2 是质数)
10-> 1010 (2 个计算置位,2 是质数)
代码
class Solution {
public:
int countPrimeSetBits(int L, int R) {
set<int>prime({2,3,5,7,11,13,17,19});
int res=0;
for(int i=L;i<=R;i++)
{
int sum=0;
for(int k=i;k;k>>=1)//将二进制中的末尾数右移
sum+=k&1;
if(prime.count(sum))
res++;
}
return res;
}
};