LeetCode 204. 计数质数

最新推荐文章于 2024-09-18 15:03:01 发布

可达鸭丶

最新推荐文章于 2024-09-18 15:03:01 发布

阅读量141

点赞数

文章标签： leetcode java

本文链接：https://blog.csdn.net/qq_20067165/article/details/109739263

版权

两种筛法

埃氏筛法

筛掉所有质数的倍数。每找到一个质数就把它所有的倍数筛掉。

class Solution {
    public int countPrimes(int n) {
        int[] primes = new int[n + 1];
        int cnt = 0;
        boolean[] st = new boolean[n + 1];
        for(int i = 2; i < n; i++){
            if(!st[i]){
                primes[cnt ++] = i;
                for(int j = i + i; j < n; j += i) st[j] = true;
            }
        }
        return cnt;
    }
}

时间复杂度

$O (n / 2 + n / 3 + n / 5 + . . .) = O (n l o g l o g n)$

线性筛

每次只拿最小质因子筛掉合数，换言之，如果一个数被筛掉了，那么一定是被它的最小质因子筛掉的。

当 $i$ 时,说明此时遍历到的 $p r i m e s [j]$ 不是i的质因子，那么只可能是此时的 $p r i m e s [j] < i$ 的最小质因子,所以 $p r i m e s [j] * i$ 的最小质因子就是 $p r i m e s [j]$ ;
当有 $i$ 时,说明 $i$ 的最小质因子是 $p r i m e s [j]$ ,因此 $KaTeX parse error: Undefined control sequence: \* at position 11: primes[j] \̲*̲ i$ 的最小质因子也就应该是 $p r i m e [j]$ ，之后接着用 $s t [p r i m e s [j + 1] * i] = t r u e$ 去筛合数时，就不是用最小质因子去更新了,因为 $i$ 有最小质因子 $p r i m e s [j] < p r i m e s [j + 1]$ ,此时的 $p r i m e s [j + 1]$ 不是 $p r i m e s [j + 1] * i$ 的最小质因子，此时就应该退出循环，避免之后重复进行筛选。

class Solution {
    public int countPrimes(int n) {
        int[] primes = new int[n + 1];
        int cnt = 0;
        boolean[] st = new boolean[n + 1];
        for(int i = 2; i < n; i++){
            if(!st[i]) primes[cnt ++] = i;
            for(int j = 0; primes[j] <= n / i; j ++) {
                st[i * primes[j]] = true;
                if(i % primes[j] == 0) break;
            }
        }
        return cnt;
    }
}