leetcode 204. 计数质数-厄拉多塞筛法（埃氏筛）、线性筛

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

示例 1：

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

示例 2：

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

输入：n = 1
输出：0

提示：

0 <= n <= 5 * 106

遍历

遍历判断[2,n]中所有整数是否为质数。

判断每个数i(i<=n)是否为质数时，只需判断[2,sqrt(i)]中每个数是否为i的因数

厄拉多塞筛法/埃氏筛

所有合数都是质数的倍数

设置数组 isPrime[i]表示数 i 是否为质数（1：是质数；0：不是质数/是合数）。for i=2从小到大遍历每个数，如果i为质数，1.质数计数+1；2.将其所有的倍数都标记为合数（除了该质数本身）。

【注意】将质数i的m倍 i*m(m>=2) 都标记为合数，这一步其实可以从该质数i的i倍 i*i 开始标记，因为质数i的m倍可视为m的i倍，当m<i时，这些数已经被标记为合数，因此无需重复标记。

class Solution {
public:
    int countPrimes(int n) {
        vector<int> isPrime(n, 1);
        int ans = 0;
        for (int i = 2; i < n; ++i) {
            if (isPrime[i]) {
                ans += 1;
                if ((long long)i * i < n) {
                    for (int j = i * i; j < n; j += i) {
                        isPrime[j] = 0;
                    }
                }
            }
        }
        return ans;
    }
};

复杂度

时间复杂度：O(n*log n*log n)

空间复杂度：O(n)，isPrime[n]用于标记每个数是否为质数

线性筛

埃氏筛仍有重复标记的情况（例如：30既是3的10倍，又是5的6倍）

使用线性筛减少重复标记，降低时间复杂度

设置数组isPrime[n]标记n个数的是否为合数

设置数组Primes[]记录当前遍历到的所有质数

线性筛：标记所有整数的质数倍（Primes[i]）为合数

埃氏筛：标记所有质数的整数倍为合数

【注意】当整数i可以整除Primes[j]时，停止对该整数的倍数的标记。因为对于任意k>j，i*Primes[k]可以被i/Primes[i]*Primes[k]标记。这种方法确保每个合数只被它的最大公约数标记。

例如：质数集：Primes[]=[2,3,5,7]，i=6，Primes[j]=3，6可以整除3。

标记i*Primes[j+1]=6*5=30时，不再用6的5倍标记，而是使用10(6/3*5)的3倍标记。

class Solution {
public:
    int countPrimes(int n) {
        vector<int> primes;
        vector<int> isPrime(n, 1);
        for (int i = 2; i < n; ++i) {
            if (isPrime[i]) {     //Primes存储遍历到的所有质数
                primes.push_back(i);
            }
            for (int j = 0; j < primes.size() && i * primes[j] < n; ++j) {
                isPrime[i * primes[j]] = 0;   //标记整数i的质数primes[j]为合数，
                if (i % primes[j] == 0) {     //如果i是质数的倍数，则不标记，
                    break;
                }
            }
        }
        return primes.size();
    }
};

复杂度

时间复杂度：O(n)

空间复杂度：O(n)，isPrime[n]和Primes[]