[SWEA] (Python) 1954.달팽이 숫자

https://swexpertacademy.com/main/code/problem/problemDetail.do?problemLevel=3&contestProbId=AV5PobmqAPoDFAUq&categoryId=AV5PobmqAPoDFAUq&categoryType=CODE&problemTitle=&orderBy=FIRST_REG_DATETIME&selectCodeLang=ALL&select-1=3&pageSize=10&pageIndex=1

SW Expert Academy

 

T = int(input())

dr = [0, 1, 0, -1]
dc = [1, 0, -1, 0]

for test_case in range(1, T+1):
    N = int(input())
    snail = [[0] * N for _ in range(N)]

    r, c = 0, 0
    dist = 0 #우, 하, 좌, 상

    for k in range(1, N*N + 1):
        snail[r][c] = k
        r += dr[dist]
        c += dc[dist]

        if r < 0 or c < 0 or r >= N or c >= N or snail[r][c] != 0:
            # 실행 취소
            r -= dr[dist]
            c -= dc[dist]
            
            # 방향 전환
            dist = (dist + 1) % 4

            # 다시 움직임
            r += dr[dist]
            c += dc[dist]

    print("#" + str(test_case))

    for row in snail:
        print(*row)