LeetCode-204-计数质数

统计所有小于非负整数 n 的质数的数量。

示例 1：

输入：n = 10
输出：4
解释：小于 10 的质数一共有 4 个, 它们是 2, 3, 5, 7

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/count-primes

解题思路

用一个长度为n的布尔数组，记录哪些数是合数
遍历数字从2到n，跳过合数，遇到素数x计数加一
然后将2x, 3x, 4x, ...这些数字标记为合数

代码

class Solution {
    public int countPrimes(int n) {
        int count = 0;
        boolean[] isComposite = new boolean[n]; // 初始默认false, 全部是素数
        for (int i = 2; i < n; i++) {
            if (isComposite[i]) { // 跳过合数
                continue;
            }
            count++; // 素数 + 1
            for (int j = 2 * i; j < n; j += i) {
                isComposite[j] = true;
            }
        }
        return count;
    }
}