LeetCode 204. Count Primes（C++版）

Description:

Count the number of prime numbers less than a non-negative number, n.

思路分析：

方法一：我们知道大于2的偶数肯定不是素数。所以遍历小于n的奇数，判断是否为素数，是，count++。

class Solution {
public:
    bool isPrime(int n){
        if(n == 1) return false;
        for(int i = 2; i <= sqrt(n); i ++){
            if(n % i == 0)
                return false;
        }
        return true;
    }
    int countPrimes(int n) {
        //首先>2的偶数肯定不是质数,只需要判断奇数
        int cnt = 0;
        if(n > 2) cnt ++;//2是质数
        for(int i = 3; i < n; i +=2 ){
            if(isPrime(i))
                cnt ++;
        }
        return cnt;
    }
};

但是这种思路超时。

方法二：

我们前面的思路可不可以优化呢？可以的——可以在判断是否为素数这个函数上改进。由于我们在主函数中传进来的参数一定为奇数，所以一定不能被偶数整除，那么可以把 i 改为从3开始每循环一次，增加2。

class Solution {
public:
    bool isPrime(int n){
        if(n == 1) return false;
        for(int i = 3; i <= sqrt(n); i += 2){//因为我们传进来的数一定不是偶数，所以可以直接只判断能都被奇数整除
            if(n % i == 0)
                return false;
        }
        return true;
    }
    int countPrimes(int n) {
        //首先>2的偶数肯定不是质数,只需要判断奇数
        int cnt = 0;
        if(n > 2) cnt ++;//2是质数
        for(int i = 3; i < n; i +=2 ){
            if(isPrime(i))
                cnt ++;
        }
        return cnt;
    }
};

这样就通过了。

方法三：厄拉多塞筛法

西元前250年，希腊数学家厄拉多塞(Eeatosthese)想到了一个非常美妙的质数筛法，减少了逐一检查每个数的的步骤，可以比较简单的从一大堆数字之中，筛选出质数来，这方法被称作厄拉多塞筛法(Sieve of Eeatosthese)。

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

其实，当你要画圈的素数的平方大于 n 时，那么后面没有划去的数都是素数，就不用继续判了。如下图：

C++代码如下：

class Solution {
public:
    /*厄拉多塞筛法*/
    int countPrimes(int n) {
        int cnt = 0;
        bool *del = new bool[n];
        //vector<int> del(n);
        del[2] = false;
        for(int i = 3; i < n; i ++){ // 2的倍数全部划去，也就是将2的倍数全部标为合数,其他全部标志为素数
            if(i % 2 == 0)
                del[i] = true;
            else 
                del[i] = false;
        }
        
        for(int i = 3; i < n; i ++){ 
            if(!del[i]){
                if(i*i > n) break; // 当前素数的平方大于n，跳出循环
                for(int j = i; i*j < n; j ++){
                    del[i*j] = true;
                }
            }
        }
        for(int i = 2; i < n; i ++){
            if(!del[i]) cnt ++;
        }
        
        delete [] del;
        return cnt;
    }
};