质数(prime number)又称素数,除了1和它本身外,不能整除以其他自然数,换句话说就是该数除了1和它本身以外不再有其他的因数;否则称为合数。最小的质数是2。
要判断一个整数N是不是质数很简单,看它是否能被2到sqrt(N)之间的整数整除即可。
def isPrime(n):
if n%2==0:
return False
for i in xrange(3,int(math.sqrt(n)+1),2):
if n%i==0:
return False
return True
不过要找出1到N之间的所有质数时,一个个的判定显然不是一个好主意。由于合数可以分解成一系列质数之积,所以1到N之间的合数都是1到sqrt(N)之间某个质数的倍数,排除这些合数,剩余的即为质数:
import math
import timeit
def findPrime(n):
a=[True]*(n+1)
a[0]=False
a[1]=False
for i in xrange(2,int(math.sqrt(n)+1)):
if a[i]:
k=i*i
while k<=n:
a[k]=False
k=k+i
if __name__=='__main__':
t=timeit.Timer('findPrime(2000000)','from __main__ import findPrime')
print t.timeit(1)
算法从2开始判断是否为质数,并排除质数的倍数,当2至i都被判断后,i+1是否为质数已很明确。
SPOJ Problem 2 Prime Generator 要求找出n至m之间的质数,其中1 <= m <= n <= 1000000000, n-m<=100000。
这种情况下建一个1000000000长度的序列就太浪费空间了,需要先找出1至sqrt(n)之间的质数,然后将n与m之间这些质数的倍数排除:
import math
def findPrime(n):
a=[True]*(n+1)
a[0]=False
a[1]=False
for i in xrange(2,int(math.sqrt(n)+1)):
if a[i]:
k=i*i
while k<=n:
a[k]=False
k=k+i
for i in xrange(2,n+1):
if a[i]:
yield i
def findPrimeBySeed(n,m):
if n==1:
n=2
seed=findPrime(int(math.sqrt(m)))
alist=[1]*(m-n+1)
for prime in seed:
if prime<n:
k=(prime-n%prime)%prime
else:
k=2*prime-n
while k<=m-n:
alist[k]=False
k+=prime
for i in xrange(m-n+1):
if alist[i]:
print i+n
if __name__=='__main__':
line=int(raw_input())
for i in xrange(line):
n,m=raw_input().split()
findPrimeBySeed(int(n),int(m))
print