[SWEA] <D2> 1954 - 달팽이 숫자

문제

달팽이는 1부터 N*N까지의 숫자가 시계방향으로 이루어져 있다.
다음과 같이 정수 N을 입력 받아 N크기의 달팽이를 출력하시오.

---

입력

가장 첫 줄에는 테스트 케이스의 개수 T가 주어지고, 그 아래로 각 테스트 케이스가 주어진다.
각 테스트 케이스에는 N이 주어진다.

출력

각 줄은 '#t'로 시작하고, 다음 줄부터 빈칸을 사이에 두고 달팽이 숫자를 출력한다.
(t는 테스트 케이스의 번호를 의미하며 1부터 시작한다.)

제약 사항

달팽이의 크기 N은 1 이상 10 이하의 정수이다. (1 ≤ N ≤ 10)

---

예제 입력

2    
3   
4

예제 출력

#1
1 2 3
8 9 4
7 6 5
#2
1 2 3 4
12 13 14 5
11 16 15 6
10 9 8 7

---

해결 방법

... 솔직히 말해서 어떤 식으로 구현을 해야 할 지 잘 모르겠었던 문제이다.  특별한 아이디어가 생각나지 않아서 무작정 구현 하는 방식으로 구현하였다. 

 

1. 시작은 y좌표를 고정하고, x좌표가 커지는 방향으로 간다. 

2. x가 범위 밖으로 가거나, 0이 아닌 숫자를 만나면 x축을 고정하고 y축을 증가시키는 방향으로 간다. 

3. y가 범위 밖으로 가거나, 0이 아닌 숫자를 만나면 y축을 고정하고, x축을 감소시키는 방향으로 간다. 

4. x가 범위 밖으로 가거나, 0이 아니 숫자를 만나면 x축을 고정하고, y축을 감소시키는 방향으로 간다. 

.... 

 

phase 라는 변수를 만들고 %4를 해나가면서 (0, 1, 2, 3) 으로 위 4단계의 페이즈를 구분한다. 

phase 는 " 2*n -1 " 만큼 반복된다. 

 

그렇게 하면 아래와 같은 길고 아름답지 못한 코드가 완성된다.... ;^;

t = int(input())
 
def check_in_range(x,y,n):
    return 0 <= x < n and 0 <= y <= n
 
for test_case in range(1, t+1):
    n = int(input())
 
    board = [[0]*n for _ in range(n)]
 
    cnt = 1
    phase = 0
    i,j = 0, -1
    while (phase != n*2 -1):
        # 오른쪽으로
        if phase % 4 == 0:
            for _ in range(n):
                j += 1
                if check_in_range(i,j,n):
                    if board[i][j] != 0:
                        j -= 1
                        break
                    else:
                        board[i][j] = cnt
                        cnt += 1
                else:
                    j -= 1
            phase += 1
        # 아래로
        elif phase % 4 == 1:
            for _ in range(n):
                i += 1
                if check_in_range(i,j,n):
                    if board[i][j] != 0:
                        i -= 1
                        break
                    else:
                        board[i][j] = cnt
                        cnt += 1
                else:
                    i -= 1
            phase += 1
        # 왼쪽으로
        elif phase % 4 == 2:
            for _ in range(n):
                j -= 1
                if check_in_range(i,j,n):
                    if board[i][j] != 0:
                        j += 1
                        break
                    else:
                        board[i][j] = cnt
                        cnt += 1
                else:
                    j += 1
            phase += 1
        # 위로
        elif phase % 4 == 3:
            for _ in range(n):
                i -= 1
                if check_in_range(i,j,n):
                    if board[i][j] != 0:
                        i += 1
                        break
                    else:
                        board[i][j] = cnt
                        cnt += 1
                else:
                    i += 1
            phase += 1
    print("#{}".format(test_case))
 
    for arr in board:
        print(*arr)

---

개선 사항?

1. phase를 통해서 이동 방향을 지정하는 것을 

          dy = [0, 1, 0, -1]
          dx = [1, 0, -1, 0]

   이런 식으로 관리했으면 좋았을 것 같다.