算法通关村 | 计算质数问题解析

素数和合数

我们看一下素数和合数的问题。素数又称为质数，素数首先要满足大于等于2，并且除了1和它本身之外，不能被任何其他自然数整除。其他数都是合数。比较特殊的是1即非素数，也非合数。2是唯一的同时为偶数和素数的数字。
有了定义，自然第一个问题就是如何判断一个正整数是否为素数。题目要求：给定一个正整数n(n<10^9)，判断它是否为素数。
基本的方式是从2开始依次与n取余测试，看看是否出现n%i==0的情况，如果出现了则说明当前的n能被i整除，所以就不是。理论上一直测试到n-1，假如都不是，那就是素数了。
而事实上不需要测试这么多，只要从2开始遍历一直到n^(1/2)就可以 ，不用执行到n-1。这个是有明确的数学证明的，我们不再赘述，如果不知道请回家问高中老师。所以实现代码就是：

boolean isPrime(int num) {
    int max = (int)Math.sqrt(num);
    for (int i = 2; i <= max; i++) {
        if (num % i == 0) {
            return false;
        }
    }
    return true;
}

基于该基础，就可以造题了，例如LeetCode204 给定整数 n ，返回 所有小于非负整数 n 的质数的数量 。

示例1：
输入：n = 10
输出：4
解释：小于 10 的质数一共有 4 个, 它们是 2, 3, 5, 7 。

这个问题几乎就是将上面的代码再套一层就行了，直接遍历：

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

埃氏筛

上面204. 计数质数 的题找素数的方法虽然能解决问题，但是效率太低，能否有效率更高一些的方法呢？
解决这个题有一个有效的方法，叫做埃氏筛，后来又产生了线性筛， 奇数筛等改进的方法。
基本思想是如果 x是质数，那么大于 x 的 xy的倍数 2x. 3x… 一定不是质数，因此我们可以从这一点入手。如下图所示:

维基百科动画

我们先选中数字2，2是素数，然后将2的倍数全部排除（在数组里将该位置标记为0就行了）。
接着我们选中数字3，3是素数，然后将3的倍数全部排除。
接着我们选择数字5，5是素数，然后将5的倍数全部排除。
接着我们选择 7，11，13一直到n，为什么 4 、6、8 、9 …不会再选择了呢？因为我们已经在前面的步骤中，将其变成0了。所以实现代码如下：

public int countPrimes(int n) {
    int[] isPrime = new int[n];
    Arrays.fill(isPrime, 1);
    int ans = 0;
    for (int i = 2; i < n; ++i) {
        if (isPrime[i] == 1) {
            ans += 1;
            if ((long) i * i < n) {
                for (int j = i * i; j < n; j += i) {
                    isPrime[j] = 0;
                }
            }
        }
    }
    return ans;
}