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

最近刷LeetCode题目的一些思路,题目信息

给定两个整数 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 不是质数)

注意:

  1. L, R 是 L <= R 且在 [1, 10^6] 中的整数。
  2. R - L 的最大值为 10000。

-------------------------------------------------------------------------------------- 

两步走,第一步先计算置位个数,第二步判断对应的个数是否是质数

1: 计算置位个数,位操作计算

private int getOneNumber(int n){
    int retNumber = 0;
    int sum = 0;
    for(int i=0; i<32; i++){
        int data = n>>i & 1;
        retNumber += data;
        sum += Math.pow(2,i) * data;
        if(sum >= n){
            break;
        }
    }
    return retNumber;
}

2:判断对应的个数是否是质数,然后得到返回值

public int countPrimeSetBits(int L, int R) {
    int retNumber = 0;
    int[] primes = new int[]{2,3,5,7,11,13,17,19,23,29,31};
    Set primeSet = new HashSet<>();
    for(int i:primes){
        primeSet.add(i);
    }
    for(int n=L; n<= R; n++){
        if(primeSet.contains(getOneNumber(n))){
            retNumber++;
        }
    }


    return retNumber;
}

 

你可能感兴趣的:(LeetCode,质数,位操作,二进制,LeetCode)