筛选法求素数

筛选法求素数

以时间换空间是常用的一种编程思想。

假设题目：求n以内的素数。

一、常规的解题思路是从小到大判断每一个数是否为素数，如下：

def checkPrime(n: int) -> bool:
	'''
	判断输入整数是否为质数
	'''
	for i in range(2, n):
		if not n % i:
			return False
	return True

def main(n: int) -> None:
	ans = [1, 2]
	for i in range(3, n+1):
		if checkPrime(i):
			ans.append(i)
	print(ans)

n = int(input('Enter a number:'))
main(n)

对于第n个数，我们需要检查n - 1个数。这段算法的事件复杂度是$O(n^2)$。

二、第二种方法是利用筛选法，如下：

def main(n: int) -> None:
	isPrime = [1] * n
	ans = [1]
	for i in range(2, n):
		if isPrime[i]:
			for j in range(2*i, n, i):
				isPrime[j] = 0
	for i in range(2, n):
		if isPrime[i]:
			ans.append(i)
	print(ans)
n = int(input('Enter a number:'))
main(n)		

我们先假设每个数都是素数，然后从小到大的从数组中划掉素数的所有倍数。这样，在遍历到大数之前，其要么是素数，要么已经在之前的循环里被划掉，所以遍历结束之后isPrime[i]表示的是i位置的数是否为素数，在遍历一遍把所有素数取出来就行了。

这样，虽然开始也是两次遍历，但是第二次遍历少了很多需要遍历的值，复杂度会变成：
$$O(n+\sum _{i=1}^{n}n/i)=O(n+nln2)=O(n).$$
当然，空间复杂度相应的上升了，不过一般空间总是够用的。