204. 计数质数

小豆丁 1年前 ⋅ 985 阅读
204. 计数质数
统计所有小于非负整数 n 的质数的数量。

 

示例 1:

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

输入:n = 0
输出:0
示例 3:

输入:n = 1
输出:0

解析

我们要给出比直接线性遍历快一点的

class Solution {
    public int countPrimes(int n) {
        //偶数都不可能是质数,我们只需要算奇数
        if(n<=2) return 0;
        if(n==3) return 1;

        //这里要把2提前算上
        int count = 1 ;
        boolean[] flag = new boolean[n];
        Arrays.fill(flag,true);
        for(int i = 3;i<n;i+=2){
            //这里我们只计算奇数
            if(flag[i]){
                //这里第一个进来的值是3,3是质数
                //如果是质数,我们就把当前数的奇数倍都置为false
                //偶数倍的都是偶数,我们不用管
                for(int j = 3;j*i<n;j+=2){
                    flag[i*j] = false;
                }

                //对于count值+1
                count++;
            }
        }
        return count;
    }
}