[SWEA - D3] 1954. 달팽이 숫자

 

 

나의 코드

# 우 -> 하 -> 좌 -> 상 
dr = [0, 1, 0, -1]
dc = [1, 0, -1, 0]
T = int(input())
for t in range(1,T+1):
    n = int(input())
    arr = [[0]*n for _ in range(n)]
    r = 0
    c = 0
    dist = 0
    for i in range(1,n*n+1):
        arr[r][c] = i
        r += dr[dist]
        c += dc[dist]
        
        if r < 0 or c < 0 or r >= n or c >= n or arr[r][c] != 0 :
            r -= dr[dist]
            c -= dc[dist]
            dist = (dist +1) % 4    
            r += dr[dist]
            c += dc[dist]
    
    print("#"+str(t))
    for x in arr:
        print(*x)

 

 

https://jennnn.tistory.com/3

[swea] 1954. 달팽이 숫자 / python 파이썬 / 2차원 배열의 인덱스 접근