算法 统计素数个数（埃氏筛选）

前言

题目链接

---

一、埃氏筛选

利用合数的概念（非素数），素数n 必然是合数，即2是素数，22，32，42都是合数。因此可以从2开始遍历，将所有的合数做上标记。

代码如下：

int countPrimes(int n){

    int count = 0;
    int *isPrime = (int *)malloc(sizeof(int) * n);
    if(!isPrime) return -1;


    //默认全部是质数,用 0 标记质数
    memset(isPrime, 0, sizeof(int) * n);

    // 2 是最小的质数
    for(int i = 2; i < n; i++)
    {
        if(!isPrime[i])
        {
            count++;
            for(int j = 2*i; j < n; j=j+i) //j就是合数的标记位: 2i, 3i, 4i, 5i, ...
            {
                isPrime[j] = 1;
            }
        }
    }

    return count;
}

---

二、代码优化

我们发现，2i, 3i, 4i, 5i,的序列会发生重复。

可以看到，遍历到3的时候，33之前的部分已经遍历过一次了；遍历到4的时候，44之前的部分也已经遍历过一次了。

因此从2开始遍历，从 22 开始做标记；遍历到 n 的时候，从 nn 开始做标记。

最终代码如下：

int countPrimes(int n){

    int count = 0;
    int *isPrime = (int *)malloc(sizeof(int) * n);
    if(!isPrime) return -1;


    //默认全部是质数
    memset(isPrime, 0, sizeof(int) * n);

    // 2 是最小的质数
    for(int i = 2; i < n; i++)
    {
        if(!isPrime[i])
        {
            count++;
            if ( i < sqrt(n)) // 即: i*i < n,防止数值溢出
            {
                for(int j = i*i; j < n; j=j+i) //j就是合数的标记位: 2i, 3i, 4i, 5i, ...
                {
                    isPrime[j] = 1;
                }
            }
        }
    }

    return count;
}

---