정올문제풀이

1707 : 달팽이사각형

파아란기쁨1 2021. 11. 28. 11:39
반응형
n = int(input())

arr = [[0 for i in range(n+1)] for j in range(n+1)]

num = 1
y=x=0
i=0
for i in range(n-1,0,-2):
    for j in range(i):
        arr[y][x]=num
        num+=1
        x+=1
    for j in range(i):
        arr[y][x]=num
        num+=1
        y+=1
    for j in range(i):
        arr[y][x]=num
        num+=1
        x-=1
    for j in range(i):
        arr[y][x]=num
        num+=1
        y-=1
    y+=1
    x+=1
if i==2 or i==0 : arr[y][x]=num
for i in range(n):
    for j in range(n):
        print(arr[i][j],end=" ")
    print()
반응형