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

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

给你两个整数 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 <= 106
• 0 <= right - left <= 104

思路

数据量最多只有 32，因此列出 32 以内的素数。逐个判断 L、R 内的数，转为二进制，统计 1 的数量，最后判断是否是素数即可。

解答

/**
 * @param {number} L
 * @param {number} R
 * @return {number}
 */
var countPrimeSetBits = function (L, R) {
  const prime = new Set([2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31]);
  let count = 0;
  for (let i = L; i <= R; i++) {
    let len = 0;
    for (const num of i.toString(2)) if (num === "1") len++;
    if (prime.has(len)) count++;
  }
  return count;
};