n=int(input())
list=[[0 for i in range(n)] for j in range(n)]
a=0
b=n//2
list[a][b]=1
for i in range(2,n**2+1):
    if a==n-1:
        a-=n-1
    if b==n-1:
        b-=n-1
    while list[a][b]!=0:
        if a!=n-1:
            a+=1
        else:
            a-=n-1
        if b!=n-1:
            b+=1
        else:
            b-=n-1
    list[a][b]=i
for i in range(n):
    for j in range(n):
        print(list[i][j],end=' ')
    print()

我在IDE上看了看我的代码，自己感觉和第二页的一篇python题解十分相似，但是同样是#1，TA的代码用了37ms,而我的代码用了“超出时间限制”的时间。甚至我在pycharm上面运行了10分钟没出结果。

有没有大佬知道哪里出了问题