[파이썬] 소용돌이(?) 코딩이 어렵네요 ㅠㅠ

코딩 문제 중에서 소용돌이(?)라는게 있던데요.

위 그림처럼 n*n 사이즈의 2차원 리스트에 시계방향으로 1부터 n^2까지 자연수를 순서대로 배열하는 문제입니다.

며칠동안 고민을 했는데 잘 해결되지 않아서 COS 시험을 앞두고 질문 드립니다 ㅠㅠ

구글링해보면 이 문제 때문에 다들 고생했다는 내용이 있네요 ㅜㅜ

우선 COS Pro 홈페이지의 모범답안의 풀이입니다.

정수 n을 받아서 n*n 사이즈의 소용돌이를 반환하는 함수입니다.

def in_range(i, j, n):
    return 0 <= i and i < n and 0 <= j and j < n

def solution(n):
    pane = [[0 for j in range(n)] for i in range(n)]
    dy = [0, 1, 0, -1]
    dx = [1, 0, -1, 0]
    ci, cj = 0, 0
    num = 1
    while in_range(ci, cj, n) and pane[ci][cj] == 0:
        for k in range(4):
            if not in_range(ci, cj, n) or pane[ci][cj] != 0:
                break
            while True:
                pane[ci][cj] = num
                num += 1
                ni = ci + dy[k]
                nj = cj + dx[k]
                if not in_range(ni, nj, n) or pane[ni][nj] != 0:
                    ci += dy[(k + 1) % 4]
                    cj += dx[(k + 1) % 4]
                    break
                ci = ni
                cj = nj
    return pane

위 코드와 비슷한 방식으로 제가 작성하려고 했던 코드인데 제대로 작동하지 않네요.

예컨대 3을 넣으면

[1,9,8]

[1,9,8]

[1,9,8]을 출력하네요.

def check(a, b, s, n): # 0 <= a, b < n이고, 이미 지나간 경로가 아님을 확인합니다.
    if 0<=a<n and 0<=b<n and s[a][b] == 0:
        return True
    else:
        return False

def sol(n):
    s=[[0]*n]*n
    da=[0, 1, 0, -1] # a는 y축입니다.
    db=[1, 0, -1, 0] # b는 x축입니다.
    a=b=0 # 현재 좌표를 의미합니다.
    cnt=1 # 1부터 n^2까지 카운트합니다.
    d=0 # da, db의 인덱스를 가리키는 변수입니다.
    while cnt<=n**2:
        s[a][b]=cnt # 현재 카운트를 s에 대입합니다.
        ta=a+da[d]
        tb=b+db[d]
        if check(ta, tb, s, n): # 움직이는 방향이 문제없는지 체크합니다.
            a=ta
            b=tb
        else: # 문제가 있으면 방향을 변경합니다.
            d=(d+1)%4
            a+=da[d]
            b+=db[d]
        cnt+=1
    return s

(a, b) 좌표를 찍어보면 틀리긴 하던데 제가 어디에서 잘못했는지 잘 모르겠네요 ㅠㅠ 

어떻게 풀어야 간명하게 해결되는지 여쭤봅니다.

스포어 회원분들께 항상 감사드립니다! ^-^

+) 한 방향으로 4 -> 3 -> 3 -> 2 -> 2 -> 1 -> 1 씩 이동하는 패턴을 활용하는 방법은 없을까요??