【Leetcode】204. Count Primes 204. 计数质数

解法

一个计算[1,n)区间内质数数量的算法

1. 首先，如果p是质数，那么p的所有倍数都要被mark成非质数，到p*i>=n为止
2. 如果某个数为p*q，但是q<p，那么这个数一定在之前被标记过，被q里的质因子标记了
3. 所以对质数p的标记从k=p*p开始，到k>n结束
4. 对于区间[sqrt(n),n)这里面的非质数来说，它们肯定有小于sqrt(n)的质因子，所以当我们遍历完[1,sqrt(n))之后，区间[sqrt(n),n)里的非质数也被标记完成了
5. 最后只需要重新遍历一遍[1,n)，记录质数个数即可

class Solution(object):
    def countPrimes(self, n):
        """
        :type n: int
        :rtype: int
        """
        invalid = set()
        for i in xrange(2,int(n**0.5)+1):
            if i not in invalid:
                for j in xrange(i**2,n,i):
                    invalid.add(j)
        return len(filter(lambda x:x not in invalid, xrange(2,n)))