PAT (Basic Level) Practice （中文）1013 数素数 python

出处：
https://pintia.cn/problem-sets/994805260223102976/problems/994805309963354112

输入格式：

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

输出格式：

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

我开始以为是输出从M到N之间的素数，其实题目要求的是输出第M个到第N个素数。
代码：

import math
m = list(map(int,input().split())) #读取输入值，m[0]为起始个素数，m[1]为终止个素数
def sushu(num):
    for i in range(3,int(math.sqrt(num)) + 1,2):    #检验这个数是否为素数。1.除到根号N就可以了，因为假设存在一个大于根号N的数能被N整除，那它肯定要乘以一个小于根号N的数才能等于N。2.任何大于2的偶数都不是素数，所以我们考虑素数的时候就可以忽略它们。
        if num % i == 0:
            return False
    return True
res = []
if m[1] > 1:
    res.append(2)
    j,k = 0,0
    while j < m[1] - 1:
        n = 3 + k * 2   #大于2的偶数一定不是素数，所以这里排除
        if sushu(n):
            res.append(n)
            j += 1
        k += 1
else:
    res.append(2)
res2 = res[m[0] - 1:]
for i in range(len(res2) - 1):
    if (i + 1) % 10 != 0:   #如果不能被10整除，就加' '，否则换行
        print(res2[i],end = ' ')
    else:
        print(res2[i])
print(res2[len(res2) - 1])  #输出最后一个数，因为最后一个不加' '

一般会通过，但有时有的测试用例会显示超时。