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

解题思路：

位运算+判断质数。
位运算计算一个数二进制所有位上1的总数，再判断其是不是质数。

代码一：

class Solution {
public:
    int countPrimeSetBits(int left, int right) {
        unordered_set<int>  prime = {2, 3, 5, 7, 11, 13, 17, 19};
        int ans = 0;
        for(int i = left; i <= right; i++)
        {
            if(prime.count(__builtin_popcount(i))) ans++;
        }
        return ans;
    }
};

技巧：unordered_set可以存储无序的键值key（这里是存质数），快速查询键值。__builtin_popcount(int)可以快速计算二进制所有位上1的总数。注意到right <= 10⁶ < 2²⁰ ，所以二进制最多有19位，19以内的质数只有2,3,5,7,11,13,17,19，所以我们可以把这几个质数存起来，判断质数时直接查询这几个数即可，省略掉计算质数的过程。

时间复杂度O(n)，空间复杂度O(1)，其中n为right-left的最大差值。

代码二：

class Solution {
public:
    bool isPrime(int x) {                 //判断是否为质数
        if(x < 2) return false;
        for(int i = 2; i*i <= x; i++)
        {
            if(x % i == 0) return false;
        }
        return true;
    }
    int bitCount(int x) {                 //计算有多少个二进制位为1
        int ans = 0;
        while(x) {
            if(x & 1) ans++;
            x >>= 1;
        }
        return ans;
    }
    int countPrimeSetBits(int left, int right) {
        int ans = 0;
        for(int i = left; i <= right; i++)
        {
            if(isPrime(bitCount(i))) ans++;
        }
        return ans;
    }
};