【leetcode-数学】计数质数

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

示例:

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

思路：

参考网上的 埃拉托斯特尼筛法 Sieve of Eratosthenes，这个算法的过程如下图所示，我们从 2 开始遍历到根号 n，先找到第一个质数 2，然后将其所有的倍数全部标记出来，然后到下一个质数 3，标记其所有倍数，以此类推，直到根号 n，此时数组中未被标记的数字就是质数。我们需要一个 n-1 长度的 boolean 型数组来记录每个数字是否被标记，注意题目说的是所有小于非负整数 n 的质数，不包括 n，然后我们用两个 for 循环来实现埃拉托斯特尼筛法。

class Solution {
    public int countPrimes(int n) {
        if(n<=2) {
            return 0;
        }
        
        boolean[] marks = new boolean[n];
        for(int i=0;i<n;i++) {
            marks[i]=true;
        }
        
        marks[0]=false;
        marks[1]=false;
        
        double limit = Math.sqrt(n);
        for(int i=2;i<=limit;i++) {
            if(marks[i]) {
                for (int j = i * 2; j < n; j += i) {
                     marks[j]=false;
                }
            }
        }
        
        int sum = 0;
        for(int j=0;j<n;++j) {
            if(marks[j]) {
                sum++;
            }
        }
    
        return sum;
    }
}