204. Count Primes [easy] Python3和C++

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

寻找n以内的素数个数。

解法一

最笨的方法是遍历到n，逐个判断，Python3代码如下：

class Solution:
    
    
    def isPrime(self, a):
        if a <=1:
            return False
        i = 2
        while i*i <= a:
            if a % i == 0:
                return False
            i += 1
        return True
    def countPrimes(self, n):
        """
        :type n: int
        :rtype: int
        """
        num = 0
        for i in range(n):
            if self.isPrime(i):
                num += 1
        return num
    

系统显示超时，又从另一篇博客中学习到厄拉多塞筛法，原理如下。

解法二

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

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

Python3版本

class Solution:
    def countPrimes(self, n):
        """
        :type n: int
        :rtype: int
        """
        if n < 3:
            return 0
        flag = [0]* n
        sq = int(n**0.5)+1
        count = 1
        for i in range(3,n,2):
            if not flag[i]:
                count += 1
                if i > sq:
                    continue
                for j in range(2*i, n, i):
                    flag[j] = 1
        return count

C++版本

#include<cmath>
class Solution {
public:
    int countPrimes(int n) {
        bool *Del = new bool[n];
        if (n < 3)
            return 0;
        for (int i = 0; i< n; i++)
            Del[i] = false;
        int count = 1;
        int sq = sqrt(n)+1;
        for (int i = 3; i < n; i += 2)
            if (!Del[i])
            {
                count += 1;
                if (i > sq)
                    continue;
                for (int j = 2*i; j < n; j += i)
                    Del[j] = true;
            }
        return count;
    }
};