统计所有小于非负整数 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;
}
}