204. Count Primes Python代码

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

Example:

Input: 10
Output: 4
Explanation: There are 4 prime numbers less than 10, they are 2, 3, 5, 7.

算法的思想就是以此通过：22,33,44,55…ii；ii+i 去排除非素数。

import math
class Solution(object):
    def countPrimes(self, n):
        """
        :type n: int
        :rtype: int
        """
        if n < 3:
            return 0
        primes = [1] * n	#列表中的值：1表示是素数，0表示不是素数
        primes[0] = primes[1] = 0
        for i in range(2, int(n ** 0.5) + 1):
            if primes[i]:
                primes[i * i: n: i] = [0] * len(primes[i * i: n: i])	#这一步就是不断剔除不是素数的值，比如2*2=4必然不是素数，再此基础上加2：4+2=6,6+2=8也都必然不是素数，因此将其primes[i]置0； 再比如3*3=9，9+3=12...也得必然不是素数。以n=10为例：i=2时，排除了4,6,8. i=3时排除了9. 至此排除了10以内的所有非素数。
        return sum(primes)
            
            

替换：
primes[i * i: n: i] = [0] * ((n-ii-1)/i+1)
效率更高。((n-ii-1)/i+1)计算非素数个数比用len快