Count Primes-Python

最新推荐文章于 2023-04-05 02:41:31 发布

原创最新推荐文章于 2023-04-05 02:41:31 发布 · 687 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#Count Primes #Python

收录于

Python 专栏收录该内容

58 篇文章

订阅专栏

本文介绍了一种在LeetCode上被高度评价的方法来计算小于n的所有素数的数量。通过两次不同的实现展示了如何利用布尔数组标记合数，并有效地跳过非素数，从而减少计算时间。

LeetCode下投票最高的操作：

class Solution:
# @param {integer} n
# @return {integer}
def countPrimes(self, n):
    if n < 3:
        return 0
    primes = [True] * n
    primes[0] = primes[1] = False
    for i in range(2, int(n ** 0.5) + 1):
        if primes[i]:
            primes[i * i: n: i] = [False] * len(primes[i * i: n: i])
    return sum(primes)

改进：

class Solution(object):
    def countPrimes(self, n):
        """
        :type n: int
        :rtype: int
        """
        if n <= 2:
            return 0
 
        prime = [True] * n
        prime[:2] = [False, False]
        for base in xrange(2, int((n - 1) ** 0.5) + 1):
            if prime[base]:
                prime[base ** 2::base] = [False] * len(prime[base ** 2::base])
        return sum(prime)