2021-10-14

计数质数

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

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

厄拉多塞筛法：
先将 2~n 的各个数放入表中，然后在2的上面画一个圆圈，然后划去2的其他倍数；第一个既未画圈又没有被划去的数是3，将它画圈，再划去3的其他倍数；现在既未画圈又没有被划去的第一个数 是5，将它画圈，并划去5的其他倍数……依次类推，一直到所有小于或等于 n 的各数都画了圈或划去为止。这时，表中画了圈的以及未划去的那些数正好就是小于 n 的素数。

public static int countPrimes(int n) {
        // 3 以下的直接返回结果
        if(n < 3) {
            return 0;
        }
        if(n == 3) {
            return 1;
        }
        int count= 0;
        boolean[] mark = new boolean[n];
        mark[2] = false;
        // 先标记2
        for (int i = 3; i < n; i++) {
            if (i % 2 == 0) {
                mark[i] = true;
            } else {
                mark[i] = false;
            }
        }
        for (int i = 3; i < n; i += 2) {
            // 之后第一个未被划去
            if (!mark[i]) {
                if (i * i > n) {
                    // 当前素数的平方大于n，跳出循环
                    break;
                }
                // 打标记（j是倍数）
                for (int j = 2; i * j < n; ++j) {
                    mark[i * j] = true;
                }
            }
        }
        // 计数
        for(int i=2;i<n;i++) {
            if(!mark[i]) {
                count ++;
            }
        }
        return count;
    }