LeetCode-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

代码Java

public int countPrimeSetBits(int left, int right) {
    int ans = 0;
    int pre = oneNum(left); // 获取left中1的个数
    for (int i = left; i <= right; i++) {
        if (isPrimeNum(pre)) ans++;
        // 获取 i 中0的位置
        int index = zeroIndex(i);
        pre += (2 - index);
    }
    return ans;
}

/**
 * @param num
 * @return 返回 num 最后一个0的位置
 */
public int zeroIndex(int num) {
    for (int i = 1; true; i++) {
        if ((num & 1) != 1) return i;
        num >>>= 1;
    }
}


/**
 * @param num 一个正整数
 * @return 是质数返回true 否则返回false
 */
public boolean isPrimeNum(int num) {
    if (num == 1) return false;
    for (int i = 2; i * i <= num; i++) {
        if (num % i == 0) return false;
    }
    return true;
}

/**
 * @param num 一个正整数
 * @return 返回一个数对应二进制数中1的个数
 */
public int oneNum(int num) {
    int count = 0;
    while (num != 0) {
        count++;
        num &= (num - 1);
    }
    return count;
}