[LeetCode 204] Count Primes(Python)

最新推荐文章于 2021-03-07 09:07:18 发布

原创最新推荐文章于 2021-03-07 09:07:18 发布 · 959 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #python

算法专栏收录该内容

55 篇文章

订阅专栏

本文介绍了一种高效的算法来计算小于非负数n的素数个数。通过使用辅助数组标记2到n的平方根之间的所有倍数，最终遍历数组以计算素数总数。此外，还提供了一个更高效的方法来判断一个数是否为素数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

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

思路

开辟一个辅助数组，依次标记 $2-\sqrt n$ 的所有倍数。最后遍历该数组，计数素数。

代码

class Solution(object):
    def countPrimes(self, n):
        """
        :type n: int
        :rtype: int
        """
        if n is None or n <= 1:
            return 0
        tmp = [True] * n
        tmp[0] = False
        tmp[1] = False

        i = 2
        while i * i < n:
            if tmp[i]:
                j = i
                while j * i < n:
                    tmp[i * j] = False
                    j += 1
            i += 1

        res = 0
        for k in tmp:
            if k:
                res += 1
        return res

复杂度分析

时间复杂度 $O(nlglgn)$ ，空间复杂度 $O(n)$ 。

如何判断一个数是否为素数

法一
遍历 $2$ 到 $（n-1）$ ，看是否有数字能整除 $n$ 。时间复杂度 $O(n)$ 。
法二
遍历 $2$ 到 $\sqrt n$ ，看是否有数字能整除 $n$ 。时间复杂度 $O(logn)$ 。

法三
大于等于5的素数一定和6的倍数相邻。
若 $x\ge1$ ， $...,6x-1,6x,6x+1,6x+2,6x+3,6x+4,6x+5,6(x+1),...$ ，除 $6x-1$ 和 $6x+1$ 都一定不是素数（因为都能找到约数）。那么我们就可以以6为步长，只判断6的倍数左右两侧的数，加快速度。

class Solution:
"""
@param: head: a ListNode
@return: a ListNode
"""

def isPrime(self, n):
    if n is None or n < 2:
        return False
    if n == 2 or n == 3:
        return True
    if n % 6 != 1 and n % 6 != 5:
        return False
    i = 5
    while i * i <= n:
        if n % i == 0 or n % (i + 2) == 0:
            return False
        i += 6
    return True