Leetcode204--计数质数

Leetcode204–计数质数

统计所有小于非负整数 n 的质数的数量。

编程语言：python

作者：黑暗主宰

邮箱：shengzhanhe@gmail.com

Leetcode204–计数质数

题目描述

原题链接：

​ https://leetcode-cn.com/problems/count-primes/ (中文)

​ https://leetcode.com/problems/count-primes/ (英文)

题目描述：

​ 统计所有小于非负整数 n 的质数的数量。

示例 1:   输入： 10   输出： 4   解释： 解释: 小于 10 的质数一共有 4 个, 它们是 2, 3, 5, 7 。

解题思路

对于本题，我刚开始用的就是普通的循环方法，先判断是否为质数，再统计整数 $n$ 范围内质数的数量。但是这种方法的时间复杂度为$O(n^2)$, 可以是python执行效率问题，直接就是TLE（Time Limit Exceeded）了。然后就google了一下，埃拉托色尼筛选法（Sieve of Eratosthenes）可以有效的解决。

下面是暴力循环的方法（注：该方法提交不能AC）：

def countPrimes(n):
	if n < 2:
		return 0
	cnt = 0
	for i in range(2, n):
		flag = 1
		for j in range(2, int(i**0.5)+1):
			if i%j == 0:
				flag = 0
				break
		if flag == 1:
			cnt += 1
	return cnt

对于判断质数的过程中，为什么要判读到$\sqrt n $呢，这里可以用反证法解释，假如$n$是合数，那么可定存在两个约数相乘等于 $n$ (除了1和本身)，我们设 $x$, $y$为大于$\sqrt n $的约数，那么$n=xy$；因为$x$, $y$都大于$\sqrt n $，所以$xy>n$,相互矛盾。所以只要判读$\sqrt n $以下的数字就可以判读是不是质数。（整体意思是，如果是一个合数，可定会出现一个小于$\sqrt n $的数和一个大于$\sqrt n $的数）。

埃拉托色尼筛选法解法：

首先放一个动态图¹，

主要思想是，假如我们有一个数字$n$，这里如图设$n=120$，我们可以知道2是一个质数，然后从2为基准，把含有2的倍数的元素全部划掉，然后剩下的数字中从3开始，3是质数，再以3为基准，把3的倍数全部去掉，一次类推，最后剩下的数就是小于$n$的所有的质数。下面的代码参考图片应该很好理解。

def countPrimes(n):
	if n <= 2:
		return 0
    # 初始化,1表示质数，0表示合数
	d = [1]*n
	d[0] = d[1] = 0
	
	for i in range(2, int(n**0.5)+1):
		if d[i] == 1:
            # 如果当前值是质数，在此基础上把对应的合数位置的flag置零
			for j in range(i+i, n, i):
				d[j] = 0
	return sum(d)

欢迎大家关注我的个人公众号，同样的也是和该博客账号一样，专注分析技术问题，我们一起学习进步

注： 文中有写错的地方，欢迎大家不吝指正！！!

---

1. https://en.wikipedia.org/wiki/Sieve_of_Eratosthenes#Algorithm_complexity ↩︎