leetcode 204-计数质数

题目描述

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

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

示例 2：
输入：n = 0
输出：0

示例 3：
输入：n = 1
输出：0

题解

题解转载自Sweetie
这是一个古老的筛素数的方法。方法如下：
初始化长度 O(n) 的标记数组，表示这个数组是否为质数。数组初始化所有的数都是质数.
从 2 开始将当前数字的倍数全都标记为合数。标记到sqrt(n) 时停止即可。

具体可以看来自维基百科的动画：

注意每次找当前素数 xx 的倍数时，是从 x^2开始的。因为如果 x > 2，那么 2*x 肯定被素数 2 给过滤了，最小未被过滤的肯定是 x^2

代码实现

class Solution {
    public int countPrimes(int n) {
        int sum = 0;
        boolean[] isPrime = new boolean[n];
        Arrays.fill(isPrime,true);
        //这块要用i*i<n而不是i<sqrt(n),因为后者会向下取整
        for(int i=2;i*i<n;i++){
            for(int j=i;j*i<n;j++){
                isPrime[i*j]=false;
            }
        }
           //因为0和1不是质数，所以此处从i=2开始
        for(int i=2;i<n;i++){
            if(isPrime[i]==true){
                sum++;
            }
        }
        return sum;
    }
}