LeetCode 762. Prime Number of Set Bits in Binary Representation

给你两个整数 left 和 right ，在闭区间 [left, right] 范围内，统计并返回计算置位位数为质数的整数个数。

计算置位位数就是二进制表示中 1 的个数。

例如， 21 的二进制表示 10101 有 3 个计算置位。

示例 1：

输入：left = 6, right = 10
输出：4
解释：
6 -> 110 (2 个计算置位，2 是质数)
7 -> 111 (3 个计算置位，3 是质数)
9 -> 1001 (2 个计算置位，2 是质数)
10-> 1010 (2 个计算置位，2 是质数)
共计 4 个计算置位为质数的数字。
示例 2：

输入：left = 10, right = 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 不是质数)
共计 5 个计算置位为质数的数字。

提示：

1 <= left <= right <= $10^6$
0 <= right - left <= $10^4$

代码

class Solution {
public:
    /*
    区间10^6 2^20 = 1024^2 > 10^6
    L和R的二进制表示最多只有二十位 可先找出二十位的质数
    */
    int countPrimeSetBits(int L, int R) {
        unordered_set<int> primes({2, 3, 5, 7, 11, 13, 17, 19});

        int res = 0;
        for(int i = L; i <= R; i++)
        {
            int s = 0;
            // k >>= 1
            for(int k = i; k; k >>= 1) s += k & 1;//每次看k的最低位是不是1 然后右移去掉最低位
            if(primes.count(s))
                res ++;
        }
        return res;
    }
};