1013 数素数 (20 分) python

ps:像素数这种东西都是有规律哦!
交第一遍可能有个点会超时 第二遍必过(这个是因为写法上的原因,可以修改(代码会变长) 但素数按照这个思路写基本没错)

import  math
a,b=map(int,input().split())
l=[2,3]
q=5
while len(l)<b:
    l+=[i for i in range(q,q+3,2) if next((t for t in range(3,int(math.sqrt(i))+1,2) if not i%t),0)==0]
    q+=6
l=l[a-1:b]
for i in range(0,len(l),10):
    print(*l[i:i+10])