题目描述:
给定两个整数 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。
解法:
class Solution {
public:
int getOnes(int num){
int res = 0;
while(num){
if(num&1){
res++;
}
num >>= 1;
}
return res;
}
int countPrimeSetBits(int L, int R) {
int res = 0;
vector<bool> flag(32, false);
vector<int> primes = {2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31};
for(int val : primes){
flag[val] = true;
}
for(int num = L; num <= R; num++){
int idx = getOnes(num);
if(flag[idx]){
res++;
}
}
return res;
}
};