Python系列问题（二）

问题

求10000000以内所有素数并输出。‪‬‪‬‪‬‪‬‪‬‮‬‫‬‮‬‪‬‪‬‪‬‪‬‪‬‮‬‫‬‪‬‪‬‪‬‪‬‪‬‪‬‮‬‫‬‫‬‪‬‪‬‪‬‪‬‪‬‮‬‪‬‮‬‪‬‪‬‪‬‪‬‪‬‮‬‪‬‮‬‪‬‪‬‪‬‪‬‪‬‮‬‪‬‮‬

素数指从大于1，且仅能被1和自己整除的整数。‪‬‪‬‪‬‪‬‪‬‮‬‫‬‮‬‪‬‪‬‪‬‪‬‪‬‮‬‫‬‪‬‪‬‪‬‪‬‪‬‪‬‮‬‫‬‫‬‪‬‪‬‪‬‪‬‪‬‮‬‪‬‮‬‪‬‪‬‪‬‪‬‪‬‮‬‪‬‮‬‪‬‪‬‪‬‪‬‪‬‮‬‪‬‮‬

要求：（1）尽量优化代码速度 （2）将判断素数的功能包装成函数实现。

方案一:

可以使用以下方法判断一个数是否为素数：从2开始到这个数的开方结束，对这个数进行取余运算，如果余数为0，说明不是素数，否则是素数。

接下来就可以实现求10000000以内所有素数的程序，代码如下：

import math

def is_prime(num):
    if num <= 1:
        return False
    for i in range(2, int(math.sqrt(num))+1):
        if num % i == 0:
            return False
    return True

if __name__ == '__main__':
    for i in range(2, 10000001):
        if is_prime(i):
            print(i)

程序中首先定义了一个判断素数的函数is_prime()，接着在主程序中循环遍历2到10000000的所有数，将每个数传入is_prime()中进行判断，如果是素数就输出该数。

方案二

可以将范围缩小为2到10000000的平方根，因为一个数如果不是素数，一定可以分解成两个数的乘积，其中一个数小于或等于它的平方根，一个数大于或等于它的平方根。代码如下：

import math

def is_prime(num):
    if num <= 1:
        return False
    for i in range(2, int(math.sqrt(num))+1):
        if num % i == 0:
            return False
    return True

if __name__ == '__main__':
    for i in range(2, int(math.sqrt(10000000))+1):
        if is_prime(i):
            print(i)

这里将范围从2到10000000改为了2到10000(10000000的平方根向上取整)。这样可以大大减少循环的次数，加快程序执行速度。

方案三

可以使用埃拉托色尼筛法（Sieve of Eratosthenes）来优化代码，该算法可以快速生成一定范围内的素数。具体步骤如下：

1. 创建一个大小为n+1的布尔数组prime[]，将里面所有元素都初始化为True。
2. 从2开始到n，对于每个i，如果prime[i]为True，则将2i，3i，4i，...都标记为False。
3. 遍历整个数组，输出所有值为True的索引。

代码如下

import math

def eratosthenes(n):
    prime = [True for i in range(n+1)]
    p = 2
    while (p * p <= n):
        if (prime[p] == True):
            for i in range(p * p, n+1, p):
                prime[i] = False
        p += 1

    for p in range(2, n):
        if prime[p]:
            print(p)

if __name__ == '__main__':
    eratosthenes(10000000)

这里将要求的范围改为10000000，同时调用eratosthenes()函数来实现筛选素数的功能，优化了代码的执行速度。