[ 파이썬(python) ] 코드업 1484 - 2차원 배열 달팽이 채우기 4-1

📍 1484 : [ 기초 - 배열 연습 ] 2차원 배열 달팽이 채우기 4-1

2차원 배열 안에 달팽이 모양으로 값을 채우는 문제인데, 사실 이 문제를 풀기 위해 1460부터 쭉 풀었다. 총 2가지 유형으로 나눠서 풀 수 있다.

⚡️ while문으로 row, column 따로따로 반복문 돌리기

1. n * m의 크기로 값이 0인 2차원 배열 arr을 선언한다.
2. row(행), column(열)을 각각 0과 -1로 초기화했는데, 이유는 1씩 증가하면서 해당 위치에 cnt가 들어가기 때문이다.
3. 그리고 중간에 행과 열의 크기를 -1씩 감소시켰는데, 반복문이 진행될때마다 채워야 하는 값의 크기가 줄어들기 때문이다. (0, 0)부터 시작하여 column에 값을 채우고 난 이후 row-1 범위로 값을 채워야하고 다시 column-1 범위로 값을 채워야하기 때문이다.
4. 채워야 하는 방향이 ➡️, ⬇️일때는 step = 1의 값으로, ⬅️, ⬆️일때는 step = -1의 값으로 바꿔줘야 해당 방향으로 값을 채워줄 수 있다.

 

⚡️ dx, dy를 사용하여 좌표를 벗어나지 않게 채우는 방법

1. n * m의 크기로 값이 0인 2차원 배열 arr을 선언한다.
2. 달팽이 배열이 시작하는 좌표를 지정하고, 방향을 지정한다.
3. R, D, L, U 순서에 맞게 dx, dy를 설정한다.
4. 만약, 범위가 벗어나면 벗어난 값을 지우고 방향을 바꾼 후 다시 진행한다.

# 1번 방법
n, m = map(int, input().split())
arr = [[0] * m for _ in range(n)]
row, column, cnt = 0, -1, 0
row_size, column_size = n, m
step = 1

while row_size and column_size:
    for _ in range(column_size):    # fill column_index
        cnt += 1
        column += step
        arr[row][column] = cnt

    row_size -= 1

    for _ in range(row_size):    # fill row_index
        cnt += 1
        row += step
        arr[row][column] = cnt

    column_size -= 1
    step = -step    # switch direction

for i in arr:
    print(*i, ' ')

# 2번 방법
n, m = map(int, input().split())
arr = [[0] * m for _ in range(n)]
x, y, direction = 0, 0

# R, D, L, U
dx = [0, 1, 0, -1]
dy = [1, 0, -1, 0]

for cnt in range(1, n*m+1):
    arr[x][y] = cnt
    x += dx[direction]
    y += dy[direction]

    if x < 0 or y < 0 or x >= n or y >= m or arr[x][y]:
        x -= dx[direction]
        y -= dy[direction]

        direction = (direction + 1) % 4
        x += dx[direction]
        y += dy[direction]

for i in arr:
    print(*i, ' ')