LeetCode每日一题--204. 计数质数（数学）

题目：跳转至 204. 计数质数
统计所有小于非负整数 n 的质数的数量。
示例 1：
输入：n = 10
输出：4
解释：小于 10 的质数一共有 4 个, 它们是 2, 3, 5, 7 。
示例 2：
输入：n = 0
输出：0
示例 3：
输入：n = 1
输出：0
提示：

• 0 <=n <= 5 * 10⁶

class Solution {
public:
    int countPrimes(int n) {
        
    }
};

思路：
想了一秒钟质数是啥，加额外一个函数辅助判断是否是质数就可以了。简单暴力愉快。

class Solution {
public:
    int countPrimes(int n) {
        int sum=0;
        for(int i=2;i<n;++i){
            if(isPrime(i))
                ++sum;
        }
        return sum;
    }
    int isPrime(int n){
        for(int i=2;i*i<=n;++i){
            if(n%i==0)
                return 0;
        }
        return 1;
    }
};

看题解埃氏筛很有点东西，该算法由希腊数学家厄拉多塞（\rm EratosthenesEratosthenes）提出，称为厄拉多塞筛法，简称埃氏筛，偷了个动图：图源自维基百科

先初始标记数组全为1，把所有数标记成质数，从2开始把当前数的倍数对应的数组值记为0，优化可以从x²开始遍历。
如上图可见：
2,3,4 ,5,6 ,7,8 ,9 ,10
2从4开始6，8，10…
3从9开始12…
4从16开始…
余下2,3,5,7均为质数，在x²前的数因为前面值的倍数已经被判断过了。

class Solution {
public:
    int countPrimes(int n) {
        vector<int> isPrime(n,1);
        int res=0;
        for(int i=2;i<n;++i){
            if(isPrime[i])
                ++res;
            if((long long) i*i<n){  //刚开始自己没加long long超出int表示范围了
                for(long long j=i*i;j<n;j+=i) //每次加上i即比上次多乘一个倍数
                    isPrime[j]=0;
            }
        }
        return res;
    }
};