Leetcode204.计数质数

题目：给定整数 n ，返回 所有小于非负整数 n 的质数的数量 。

原解题思路：

定义一个标志数flag，初始值为1，再定义一个计数器k，初始值为0。首先，如果n<2,那么不可能为质数，返回0。用双层for循环对所有小于n的数进行质数判断，如果该数对大于二的值取模为0，即该数字不是质数，则将flag赋值为0。在内层循环结束后，如果flag值为1，则计数器k+1；循环结束后，返回k的值。

代码如下：

int countPrimes(int n){
    int flag=1,k=0;
    if(n<2) return 0;
    for(int i=2;i<n;i++){
         flag=1;
        for(int j=2;j<=sqrt(i);j++){
            if(i%j==0){
                flag=0;
                break;
            }
        }
        if(flag==1) {
            k++;
        }
    }
    return k;
}

提交代码，显示超出时间限制。

分析：使用了简单的试除法来检查每个数是否为质数，这种方法对于较大的 n 来说效率很低。所以需优化判断质数的算法，通过查看题目提示3，了解到埃拉托色尼筛法。即要得到自然数n以内的全部质数，必须把不大于根号n的所有质数的倍数剔除，剩下的就是质数。

思路：创建一个bool类型的数组isPrime，大小为n，用来标记从0到n-1的每个数是否为质数。初始化数组，将所有元素设为true，表示假设所有数都是质数。因为0和1不是质数，所以isPrime[0]和isPrime[1]设为false。使用一个循环，从2开始，直到p * p（p的平方）小于n。用来找到所有小于sqrt(n)的质数，并用它们来标记它们的倍数为非质数。如果p是一个质数，那么将p的所有倍数（从p * p开始，每次+p）在isPrime数组中标记为false。再使用循环循环，从2开始，直到n-1，遍历isPrime数组。对于数组中的每个true值，我们将质数计数器count增加1。循环结束后，count变量就包含了小于n的质数数量。最后，返回这个值作为结果。

代码：

int countPrimes(int n) {
    if (n < 2)
        return 0;

    bool isPrime[n];
    for (int i = 0; i < n; i++) {
        isPrime[i] = true;
    }
    isPrime[0] = isPrime[1] = false;

    int count = 0;
    for (int p = 2; p * p < n; p++) {
        if (isPrime[p]) {
            // 标记所有p的倍数为非质数
            for (int i = p * p; i < n; i += p) {
                isPrime[i] = false;
            }
        }
    }

    // 计数剩余的质数
    for (int i = 2; i < n; i++) {
        if (isPrime[i]) {
            count++;
        }
    }

    return count;
}