（位运算）leetcode762:二进制表示中质数个计算置位

题目

给定两个整数 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。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/prime-number-of-set-bits-in-binary-representation

思路

1.因为给出整数范围是10 ^ 6，所以可以先把该范围的质数枚举出来保存在一个哈希表中。因为2^20 >10 ^ 6,所以枚举下20以内的质数
2.统计每个数的计算置位个数，判断是不是质数

错误代码

class Solution {
public:
    int countPrimeSetBits(int L, int R) {
        
        unordered_set<int> primes={2,3,5,7,11,13,17,19};
        int cnt=0;
        for(int i=L;i<=R;i++)
        {
            int k=0;
            **while(i)
            {
               k+=i&1;
               i>>1;
            }**
            if(primes.count(k))
            {
                cnt++;
            }
        }
        return cnt;

    }
};

错误原因

**处就是错误地方，如果直接使用i，那么i值就会始终发生变化，导致计算超时，此时计算的也不是每个数的计算质位，应该重新设置一个变量，来保存每次遍历的值

正确代码

class Solution {
public:
    int countPrimeSetBits(int L, int R) {
        
        unordered_set<int> primes={2,3,5,7,11,13,17,19};
        int cnt=0;
        for(int i=L;i<=R;i++)
        {
            int k=0;//置位的个数
            for(int j=i;j;j>>=1)//计算i的置位个数
            {
                k+=j&1;//一个数和1做位与运算，该数的二进制最右一位如果为1，其结果为1；该数的二进制最右一位如果为0，其结果为0
            }
            if(primes.count(k))//判断key为元素k是否在primes中
            {
                cnt++;
            }
        }
        return cnt;

    }
};

备注

1.set的count函数作用是返回key为元素k的个数，返回结果有两种——1和0，返回1表示存在元素k，返回0表示不存在元素k；（所以通常用于判断set中是否存在元素k）
2.一个数和1做位与运算，该数的二进制最右一位如果为1，其结果为1；该数的二进制最右一位如果为0，其结果为0；
3.将一个数右移一位，相当于除以2，也相当于去掉该二进制的最右一位；
4.判断一个数的二进制中1的个数，可以①先和1做位与运算，②然后将该数右移一位，重复①②,直到该数为0；
5.unordered 基于哈希表，使用数组存取key值