PAT 乙级1007 python3运行超时

[luoyanhanhan]

我的代码：

def Prime_list(n):
    list = []
    flag = [1]*(n+2)
    p = 2
    while p <= n:
        list.append(p)
        for i in range(2 * p, n + 1, p):
            flag[i] = 0
        while 1:
            p += 1
            if flag[p] == 1:
                break
    return list

if __name__ == "__main__":
    N = int(input())
    list = Prime_list(N)
    count = 0
    for i in list[:-1]:
        if list[list.index(i)+1] - i == 2:
            count += 1
    print(count)

运行结果：

请大佬们指出我代码里面的问题，最好还能给出改进意见

[冯斯特罗]

其实我没看你的业务逻辑.
但是这里是一个死循环.

while p <= n:
        list.append(p)
        for i in range(2 * p, n + 1, p):
            flag[i] = 0
        while 1:
            p += 1
            if flag[p] == 1:
                break

没有break外面的大循环,同时P不变N也不变.

[luoyanhanhan]

引用 1 楼 chaleaoch_gmail 的回复:

其实我没看你的业务逻辑.
但是这里是一个死循环.

while p <= n:
        list.append(p)
        for i in range(2 * p, n + 1, p):
            flag[i] = 0
        while 1:
            p += 1
            if flag[p] == 1:
                break

没有break外面的大循环,同时P不变N也不变.

p是有变的，内循环的p在不断自增，遇到下一个flag[p]==1的时候跳出，将列表里p的倍数的位置置0，用的是埃氏筛选法求素数