【leetcode】204.Count Primes(python)

最新推荐文章于 2022-12-25 22:37:54 发布

山间wu

最新推荐文章于 2022-12-25 22:37:54 发布

阅读量301

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/qq_42011358/article/details/83542331

版权

leetcode 专栏收录该内容

29 篇文章 0 订阅

订阅专栏

题目描述
Count the number of prime numbers less than a non-negative number, n.
计算比一个非负数n小的所有质数的个数。

思路
采用厄尔多塞筛法。

代码

class Solution:
    def countPrimes(self, n):
        """
        :type n: int
        :rtype: int
        """
        if n <3:
            return 0

        digits = [1]*n
        digits[0] = digits[1] = 0

        for i in range(2, int(n**0.5)+1):
            if digits[i] == 1:
                for j in range(i+i, n, i):
                    digits[j] = 0

        return sum(digits)