Leetcode762-二进制表示中质数个计算置位

题目描述

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