PAT乙级 1013 数素数 (python)

1013 数素数 (20 分)

令 P_i表示第 i 个素数。现任给两个正整数 M≤N≤10⁴，请输出 P_M 到 P_N 的所有素数。

输入格式：
输入在一行中给出 M 和 N，其间以空格分隔。

输出格式：
输出从 P_M 到 P_N 的所有素数，每 10 个数字占 1 行，其间以空格分隔，但行末不得有多余空格。

输入样例：

5 27

输出样例：

11 13 17 19 23 29 31 37 41 43
47 53 59 61 67 71 73 79 83 89
97 101 103

python代码：

import math
r = input().split()
M = int(r[0])
N = int(r[1])
def isPrime(p, ans):
    n_sqrt = int(math.sqrt(p))+1
    for i in ans:
        if i >= n_sqrt:
            return True
        if p%i == 0:
            return False
    return True
ans = []
ans_len = 1
p = 2
while ans_len <= N:
    if isPrime(p,ans):
        ans.append(p)
        ans_len += 1
    p += 1
for i in range(M-1,N,10):
    print(' '.join(map(str,ans[i:i+10])))

参考链接：https://github.com/liuchuo/PAT/blob/master/BasicLevel_Python/1013%20%E6%95%B0%E7%B4%A0%E6%95%B0%20%EF%BC%8820%20%E5%88%86).py