leetcode204:计算质数

思想:

判断n是否小于2,若是则返回0(由于0和1都不是质数)。定义一个dp存放标志位,若为值1则标志是质数,若为值0则标志不是质数。dp[0]=0和dp[1]=0代表数字0和1都不是质数。循环函数helper的功能是给比n小的2的倍数自己除外(例如4,6,8...),3的倍数,5的倍数,7的倍数等等质数的倍数赋值0(代表他们都不是质数)。当其中一个质数的倍数大于n则结束循环。然后计算dp中1的个数即为质数的量。

class Solution:
    def countPrimes(self, n):
        """
        :type n: int
        :rtype: int
        """
        def helper(n, dp):
            for i in range(2, n):
                if dp[i] == 1:
                    k = i * i
                    if k >= n:
                        return
                    while k < n:
                        dp[k] = 0
                        k += i

        if n < 2:
            return 0
        dp = [1] * n
        dp[0] = 0
        dp[1] = 0
        helper(n, dp)
        return sum(dp)

猜你喜欢

转载自blog.csdn.net/weixin_43160613/article/details/83750772