python习题-筛法求素数

def sieve(n, m):
    """
    输入两个正整数n和m，返回[n,m]范围内的所有素数的列表
    """
    # 初始化一个长度为m-n+1的列表，用于标记数值是否为素数
    is_prime = [True] * (m - n + 1)
    # 如果n为1，则将1标记为合数
    if n == 1:
        is_prime[0] = False
    # 从2开始，将所有能被2整除的数标记为合数
    for i in range(2, int(m ** 0.5) + 1):
        for j in range(max(i ** 2, (n + i - 1) // i * i), m + 1, i):
            if j >= n and is_prime[j - n]:
                is_prime[j - n] = False
    # 将所有未被标记为合数的数加入素数列表
    primes = [i + n for i in range(m - n + 1) if is_prime[i]]
    return primes


# 输入n和m
n, m = map(int, input("请输入n和m(1 < n < m < 1000):").split())
if not 1 < n < m < 1000:
    print("输入有误，请重新输入")
    n, m = map(int, input("请输入n和m(1 < n < m < 1000):").split())
else:
    print("该范围内的所有素数为：", end='')
    print(sieve(n, m))