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

题

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

解题思路

• 0 1不是质数
• 由于不超int范围 直接计算0到32有几个质数就好
• 直接初始化好判断数组更省时间

代码

class Solution {
    private val judge = BooleanArray(35)

    init {
        judge[0] = false
        judge[1] = false
        judge[2] = true
        for (i in 3..32) {
            var isPrime=true
            for (j in 2..Math.sqrt(i.toDouble()).toInt()){
                if (i%j==0){
                    isPrime=false
                    break
                }
            }
            judge[i]=isPrime
        }
    }

    fun countPrimeSetBits(left: Int, right: Int): Int {
        var ans=0
        for (i in left..right){
            var oneCount=0
            val iBinary=i.toString(2)
            for (c in iBinary) {
                if (c=='1'){
                    oneCount++
                }
            }
            if (judge[oneCount]){
                ans++
            }
        }
        return ans
    }
}