Leetcode 204.Count Primes 素性检测

204.Count Primes

1. 问题描述
   Count the number of prime numbers less than a non-negative number, n.
   翻译：计算小于一个非负数n的素数的数量
2. 解题思路
   （1）最简单的方法–暴力求解
   一个函数 countPrimes 算小于非负数n的素数的数量，同时在里面的循环里调用另一个bool型的函数 isPrime ，检测单个数字是否是素数。

public int countPrimes(int n) {
   int count = 0;
   for (int i = 1; i < n; i++) {
      if (isPrime(i)) count++;
   }
   return count;
}

private boolean isPrime(int num) {
   if (num <= 1) return false;
   // Loop's ending condition is i * i <= num instead of i <= sqrt(num)
   // to avoid repeatedly calling an expensive function sqrt().
   for (int i = 2; i * i <= num; i++) {//注意用i*i来代替sqrt函数，因为其花销巨大
      if (num % i == 0) return false;
   }
   return true;
}

但是！这样并不能在对运算时间苛刻的Leetcode的测试，即使其时间花销已经为O(n^1.5)。

（2）改进思路–Eratosthenes筛法
如图：

(i). 我们从一个包含n个数字的表开始。 让我们看看第一个数字，我们知道2的所有倍数都不能是素数，所以我们将它们标记为非素数。 然后我们看下一个数字，3.类似地，3的所有倍数，如3×2 = 6，3×3 = 9，…不能是素数，因此我们也标记它们。 现在我们看下一个数字，4，已经标记了。 这告诉你什么？ 你应该标记4的所有倍数吗？
(ii). 4不是素数，因为它可以被2整除，这意味着4的所有倍数也必须被2整除，并且已经被标记。 因此，我们可以立即跳过4并转到下一个数字5.现在，5的所有倍数，例如5×2 = 10,5×3 = 15,5×4 = 20,5×5 = 25，… 可以标记掉。 这里有一个轻微的优化，我们不需要从5×2 = 10开始。我们应该从哪里开始标记？
(iii).事实上，我们可以从5×5 = 25开始标记5的倍数，因为5×2 = 10已经被2的倍数标记，类似地5×3 = 15已经被3的倍数标记。因此， 如果当前数字是p，我们可以总是标记p的倍数，从p2开始，然后以p的增量：p2 + p，p2 + 2p，…现在应该是终止循环条件？
(iv).终止循环条件可以是p <√n，因为所有非素数≥√n必须已经标记。 当循环终止时，表中未标记的所有数字都是素数。
3. 完整测试代码

#include<iostream>
using namespace std;


int countPrimes(int n)
{
    int count=0;
    bool *a = new bool[n]();//开一个bool型的数组，非负数标记为true
    for (int i = 2; i*i < n; i++)
    {
        if (!a[i])//如果a[i]是素数，则对a[i]的倍数（至少是i倍，小于i倍的已经被标记）进行标记
        {
            for (int j = i; i*j < n; j++)//j从i开始，减小时间开销
            {
                a[i*j] = true;
            }
        }
    }
    for (int i = 2; i < n; i++)
    {
        if (!a[i])//如果a[i]未被标记，则素数数量加1
            count++;
    }
    return count;
}

int main()//测试用例
{
    int n;
    cout << "Please input a non-negative number n:\n";
    while (cin >> n)
    {
        cout <<"The number of primes less than n is:\n" <<countPrimes(n)<<
            "\n--------------------------------------\n"<<"Please input a non-negative number n:\n";
    }
    return 0;
}

测试结果图：