Count Primes —— Leetcode

最新推荐文章于 2020-03-07 22:19:14 发布

原创最新推荐文章于 2020-03-07 22:19:14 发布 · 2.9k 阅读

2 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

48 篇文章

订阅专栏

本文介绍了一种基于埃拉托斯特尼筛法的算法，用于计算小于非负整数n的所有质数的数量。通过使用布尔数组标记合数，并迭代去除它们，最终得到所需质数的总数。

Description:

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

click to show more hints.

References:

How Many Primes Are There?

Sieve of Eratosthenes

参考References第二个链接，wikipedia上的动图，筛选算法，最后得出素数个数，算法如下：

注意int *array = new int[n]和int *array = new int[n]()的区别，前者不对数组初始化，后者初始化为0；

另外一个需要注意的地方是两个for循环的终止条件，不大于sqrt(n)和n；

我的源码如下：

class Solution {
public:
    int countPrimes(int n) {
        if(n<=2)
            return 0;
        int *array = new int[n]();
        int tmp, sum, count = n-2;
        for(int i=2; i<=(int) sqrt(n); i++)
        {
            if(!array[i])
            {
                tmp = i*i;
                for(int j=0; (sum=tmp+i*j)<n; j++)
                {
                    if(!array[sum])
                    {
                        array[sum] = true;
                        count--;
                    }
                }
            }
        }
        delete []array;
        
        return count;
    }
};