P5731 【深基5.习6】蛇形方阵 python解法

可以选择打表也可以选择模拟

打表：

n=int(input())
# arr=[[0 for _ in range(n)] for _ in range(n)]
if n==1:
    print("  1")
elif n==2:
    print("  1  2\n  4  3")
elif n==3:
    print("  1  2  3\n  8  9  4\n  7  6  5")
elif n==4:
    print("  1  2  3  4\n 12 13 14  5\n 11 16 15  6\n 10  9  8  7")
elif n==5:
    print("  1  2  3  4  5\n 16 17 18 19  6\n 15 24 25 20  7\n 14 23 22 21  8\n 13 12 11 10  9")
elif n==6:
    print("  1  2  3  4  5  6\n 20 21 22 23 24  7\n 19 32 33 34 25  8\n 18 31 36 35 26  9\n 17 30 29 28 27 10\n 16 15 14 13 12 11")
elif n==7:
    print("  1  2  3  4  5  6  7\n 24 25 26 27 28 29  8\n 23 40 41 42 43 30  9\n 22 39 48 49 44 31 10\n 21 38 47 46 45 32 11\n 20 37 36 35 34 33 12\n 19 18 17 16 15 14 13")
elif n==8:
    print("  1  2  3  4  5  6  7  8\n 28 29 30 31 32 33 34  9\n 27 48 49 50 51 52 35 10\n 26 47 60 61 62 53 36 11\n 25 46 59 64 63 54 37 12\n 24 45 58 57 56 55 38 13\n 23 44 43 42 41 40 39 14\n 22 21 20 19 18 17 16 15")
elif n==9:
    print("  1  2  3  4  5  6  7  8  9\n 32 33 34 35 36 37 38 39 10\n 31 56 57 58 59 60 61 40 11\n 30 55 72 73 74 75 62 41 12\n 29 54 71 80 81 76 63 42 13\n 28 53 70 79 78 77 64 43 14\n 27 52 69 68 67 66 65 44 15\n 26 51 50 49 48 47 46 45 16\n 25 24 23 22 21 20 19 18 17")

模拟：

大循环为输入的值，内部的循环分别为右，下，左，上，注意每次步骤后稍微调整一下行列，使之满足下个循环的条件；

同时，还要注意输出格式，每个数字都占三个子符，中间不需要空格（因为三个字符自然而然就隔开）

n=int(input())
arr=[[0 for _ in range(n)] for _ in range(n)]
k=1
i,j=0,0
while k<=n*n:
#注意边界同时判断这个位置有没有被遍历过
    while (j<n and arr[i][j]==0): #向右
        arr[i][j]=k
        k+=1
        j+=1 
    j-=1  #因为上一次循环j+1了导致不满足循环条件，所以在这里修正一下
    i+=1 #当前位置已经填完了，往下一行走
    while (i<n and arr[i][j]==0): #向下
        arr[i][j]=k
        k+=1
        i+=1
    i-=1
    j-=1
    while j>=0 and arr[i][j]==0: #向左
        arr[i][j]=k
        k+=1
        j-=1
    j+=1
    i-=1
    while i>0 and arr[i][j]==0: #向上
        arr[i][j]=k
        k+=1
        i-=1
    i+=1 
    j+=1       
for v in range(n):
    for s in range(n):
        if arr[v][s]<10:
            arr[v][s]=str(f'  {arr[v][s]}') #两空格
        else:
            arr[v][s]=str(f' {arr[v][s]}') #单空格
        print(arr[v][s],end="")
    print("")