PTA 1013 数素数(Python版)

令 Pi​ 表示第 i 个素数。现任给两个正整数 M≤N≤104，请输出 PM​ 到 PN​ 的所有素数。

输入格式：

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

输出格式：

输出从 PM​ 到 PN​ 的所有素数，每 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

提交结果：

代码： 

import math


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


a, b = input().split()
l = [2]
s = 3
while len(l) < int(b):
    if sushu(s) == True:
        l.append(s)
    s += 2
l = l[int(a) - 1:]
index1 = 1
for i in l[:-1]:
    if index1 % 10 == 0:
        print(i)
    else:
        print(i, end=' ')
    index1 += 1
print(l[-1], end='')