[프로그래머스] Lv.0 정수를 나선형으로 배치하기 (Python)

https://school.programmers.co.kr/learn/courses/30/lessons/181832

프로그래머스

문제 설명

양의 정수 n이 매개변수로 주어집니다. n × n 배열에 1부터 n2 까지 정수를 인덱스 [0][0]부터 시계방향 나선형으로 배치한 이차원 배열을 return 하는 solution 함수를 작성해 주세요.

제한사항

• 1 ≤ n ≤ 30

입출력 예

입출력 예 설명

입출력 예 #1
예제 1번의 n의 값은 4로 4 × 4 배열에 다음과 같이 1부터 16까지 숫자를 채울 수 있습니다.

따라서 [[1, 2, 3, 4], [12, 13, 14, 5], [11, 16, 15, 6], [10, 9, 8, 7]]를 return 합니다.

입출력 예 #2
예제 2번의 n의 값은 5로 5 × 5 배열에 다음과 같이 1부터 25까지 숫자를 채울 수 있습니다.

따라서 [[1, 2, 3, 4, 5], [16, 17, 18, 19, 6], [15, 24, 25, 20, 7], [14, 23, 22, 21, 8], [13, 12, 11, 10, 9]]를 return 합니다.

---

문제 풀이

1. 변수 설정
   • n X n 의 빈 2차원 배열 초기화
   • 현재 위치를 나타내는 (x, y) 좌표 초기화
   • 방향을 나타내는 변수 초기화 ( 오른쪽, 아래, 왼쪽 , 위 순서로 이동)
   • 현재까지 채운 숫자를 나타내는 변수 초기화
2. 나선형 채우기
   • 현재 방향으로 이동하면서 배열 채워나감
   • 배열의 끝에 도달하거나 이미 채워진 곳을 만나면 방향 변경
   • 모든 숫자가 배열에 채워질 때까지 2번 과정 반복
3. 완성된 배열 반환

 

[Python]

def solution(n):
    
    arr = [[0] * n for _ in range(n)]
    x, y = 0, 0 # 현재 위치
    num = 1
    dir_idx = 0
    directions = [(0, 1), (1, 0), (0, -1), (-1, 0)]  # 오른쪽, 아래, 왼쪽, 위
   
    while num <= (n * n):
        arr[x][y] = num
        num += 1
        
        next_x = x + directions[dir_idx][0]
        next_y = y + directions[dir_idx][1]
        
        if (0 <= next_x < n and 0 <= next_y < n and arr[next_x][next_y] == 0):
            x, y = next_x, next_y
        else:
            dir_idx = (dir_idx + 1) % 4
            x += directions[dir_idx][0]
            y += directions[dir_idx][1]
        
    return arr