首页 > 其他 > 详细

[leetcode]204. Count Primes

时间:2017-03-05 20:46:57      阅读:205      评论:0      收藏:0      [点我收藏+]

统计小于n的素数的个数

如果判断每一个数是不是素数的话,时间会超时,如下

 1 class Solution(object):
 2     def countPrimes(self, n):
 3         """
 4         :type n: int
 5         :rtype: int
 6         """
 7         c = 0
 8         for i in range(2,n):
 9             if self.isprime(i):
10                 c += 1
11         return c
12     
13     def isprime(self,n):
14         n = int(n**0.5)+1
15         for i in range(2,n):
16             if not n%i:
17                 break
18         else:
19             return True
20         return False

题意要统计素数的个数,那么可以换一种思路,每发现一个素数i,那么i^2,i^2+i...都肯定不是素数

 1 class Solution(object):
 2     def countPrimes(self, n):
 3         """
 4         :type n: int
 5         :rtype: int
 6         """
 7         if n<3:
 8             return 0
 9         flag = [True]*n
10         flag[0],flag[1] = False,False
11         for i in range(2,int(n**0.5+1)):
12             if flag[i]:
13                 flag[i*i:n:i] = [False]*len(flag[i*i:n:i])
14         return sum(flag)

 

[leetcode]204. Count Primes

原文:http://www.cnblogs.com/fcyworld/p/6506429.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!