Pat乙级1013题——数素数(Python)一个测试点超时

P i 表示第 i 个素数。现任给两个正整数 M N 10 4 ​ ,请输出 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

代码如下,有一个测试点超时,但我没有找到原因,如有高手,希望能帮忙解答一下。

from math import sqrt
def getPrime():
    input = map(int, raw_input().split(' '))
    start = input[0]
    end = input[1]
    result = []
    def is_prime(n):
        if n == 1:
            return False
        for i in range(2, int(sqrt(n))+ 1):
            if n % i == 0:
                return False
        return True

    number = 1
    count = 1
    while count <= end:
        if is_prime(number):
            if count >= start:
                result.append(str(number))
            count += 1
        number += 1
    meg = ''
    for i in range(len(result)):
        if (i+1) % 10 == 0 and i < (len(result)-1):
            meg = meg + result[i]+'\n'
        elif i < (len(result)-1):
            meg = meg + result[i]+' '
        else:
            meg = meg + result[i]
    print meg

if __name__ == '__main__':
    getPrime()

猜你喜欢

转载自blog.csdn.net/coder_wu/article/details/82051668