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

1) 문제 설명

• 양의 정수 n이 매개변수로 주어집니다.

 

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

 

2) 제한사항

• 1 ≤ n ≤ 30

 

3) 입출력 예제

n    result
4    [[1, 2, 3, 4], [12, 13, 14, 5], [11, 16, 15, 6], [10, 9, 8, 7]]
5    [[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]]

 

4) 입출력 예제 설명

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

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

따라서 [[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까지 숫자를 채울 수 있습니다.

0    1    2    3    4    5
1    16    17    18    19    6
2    15    24    25    20    7
3    14    23    22    21    8
4    13    12    11    10    9

따라서 [[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 합니다.

 

5) 풀이

def solution(n):
    # 좌표 및 방향 정보
    x_idx, y_idx = 0, 0
    direction = "right"

    # 배열 초기화 (0으로)
    L = [[0 for _ in range(n)] for __ in range(n)]

    # n x n 반복
    for num in range(1, n * n + 1):

        # 값 대입
        L[x_idx][y_idx] = num

        # 방향이 오른쪽인 경우
        if direction == "right":

            # 해당 조건 만족 시 방향 전환, 아니면 인덱스 증가
            if y_idx + 1 == n or L[x_idx][y_idx + 1] != 0:
                direction = "down"
                x_idx += 1
            else:
                y_idx += 1

        # 방향이 아래쪽인 경우
        elif direction == "down":

            # 해당 조건 만족 시 방향 전환, 아니면 인덱스 증가
            if x_idx + 1 == n or L[x_idx + 1][y_idx] != 0:
                direction = "left"
                y_idx -= 1
            else:
                x_idx += 1

        # 방향이 왼쪽인 경우
        elif direction == "left":

            # 해당 조건 만족 시 방향 전환, 아니면 인덱스 증가
            if L[x_idx][y_idx - 1] != 0:
                direction = "top"
                x_idx -= 1
            else:
                y_idx -= 1

        # 방향이 위쪽인 경우
        elif direction == "top":

            # 해당 조건 만족 시 방향 전환, 아니면 인덱스 증가
            if L[x_idx - 1][y_idx] != 0:
                direction = "right"
                y_idx += 1
            else:
                x_idx -= 1
    return L

 

[Reference]

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

프로그래머스