PAT乙级1013数素数-python实现(20分)

先贴个图：

一开始判断素数的时候，只是循环到数的开平方，最后测试点4超时，只拿了19分，下面是19分的代码，最后有20分的代码。

import math
m, n = map(int, input().split())
lst = []

def isPrime(x):# 判断素数
    if x <= 1:
        return False
    for i in range(2, int(math.sqrt(x))+1):
        if x % i == 0:
            return False

    return True
# 素数表
for i in range(1, 1000001):    # 因为m<=n<=10^4, 所以循环范围可以大一点
    if isPrime(i) == True:
        lst.append(i)
    if len(lst) >= n:   # 限制到找到第n个素数，就结束循环
        break
count = 0
for i in range(m - 1, len(lst)):
    count += 1
    #增加条件限制，使每行末尾都没有空格
    print(lst[i], end=' ' if i != len(lst) - 1 and count % 10 != 0 else '')
    # i != len(lst) - 1使最后一行最后一个数后面没有空格
    # count % 10 != 0 使每行的最后以个没有空格
    if count % 10 == 0:
        print('')

后来感觉应该是判断素数的时候太慢了，在判断素数的时候又增加了判断这个数是否是6n+1或者6n-1；

即 (n+1)%6 和 (n-1)%6 是否等于0.

因为所有的素数，除了2、3，都在6的附近，要么是6-1，要么是6+1
6n(+0)：一定可以被3整除
6n+2：一定可以被2整除
6n+3：一定可以被3整除
6n+4：一定可以被2整除
只有6n+1 和 6n-1(6n+5)是不确定的。

所以判断素数的时候改为如下代码即可：

def isPrime(x):
    if x <= 3:
        return x >= 2  #如果是2、3，返回True
    else:
        if (x + 1) % 6 != 0 and (x - 1) % 6 != 0:
            return False
    for i in range(2, int(math.sqrt(x))+1):
        if x % i == 0:
            return False

完整20分代码：

import math
m, n = map(int, input().split())
lst = []

def isPrime(x):# 判断素数
    if x <= 3:
        return x >= 2
    else:
        if (x + 1) % 6 != 0 and (x - 1) % 6 != 0:
            return False
    for i in range(2, int(math.sqrt(x))+1):
        if x % i == 0:
            return False

    return True
# 素数表
for i in range(1, 1000001):   # 因为m<=n<=10^4, 所以循环范围可以大一点
    if isPrime(i) == True:
        lst.append(i)
    if len(lst) >= n:   # 限制到找到第n个素数，就结束循环
        break
count = 0
for i in range(m - 1, len(lst)):
    count += 1
    #增加条件限制，使每行末尾都没有空格
    print(lst[i], end=' ' if i != len(lst) - 1 and count % 10 != 0 else '')
    # i != len(lst) - 1使最后一行最后一个数后面没有空格
    # count % 10 != 0 使每行的最后以个没有空格
    if count % 10 == 0:
        print('')

对于输出格式的要求：

每 10 个数字占 1 行，其间以空格分隔，但行末不得有多余空格。

for i in range(m - 1, len(lst)):
    count += 1
    #增加条件限制，使每行末尾都没有空格
    print(lst[i], end=' ' if i != len(lst) - 1 and count % 10 != 0 else '')
    if count % 10 == 0: #每10个数字一行，count等于10的时候另起一行
        print('')

i != len(lst) - 1 使最后一行最后一个数后面没有空格

count % 10 != 0 使每行的最后一个没有空格

可以把else后面的' '改成'#'验证一下

for i in range(m - 1, len(lst)):
    count += 1
    #增加条件限制，使每行末尾都没有空格
    print(lst[i], end=' ' if i != len(lst) - 1 and count % 10 != 0 else '#')
    if count % 10 == 0: #每10个数字一行，count等于10的时候另起一行
        print('')

结果如下：

PAT乙级1013数素数-python实现(20分)

猜你喜欢