[프로그래머스] 정수를 나선형으로 배치하기 파이썬

문제 설명

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

---

제한사항

• 1 ≤ n ≤ 30

---

입출력 예

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]]

---

입출력 예 설명

입출력 예 #1

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

  행 \ 열	0	1	2	3
  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
  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 합니다.

제출 답안 1

• if문을 사용하지 않기 위하여 combi라는 사전을 만들어서 r,d,l,u에 값들을 지정하였다.
• 직관적이지 못해서 시간이 좀 걸린다.

COL = "col"
ROW = "row"
NXT = "nxt"
COP = "compare"
EU = "equal_number"


def solution(n):
    if n == 1:
        return [[1]]
    answer = [[0] * n for _ in range(n)]
    y, x = 0, 0
    direct = "r"
    combi = {
        "r": {COL: 1, ROW: 0, NXT: "d", COP: "x", EU: n - 1},
        "d": {COL: 0, ROW: 1, NXT: "l", COP: "y", EU: n - 1},
        "l": {COL: -1, ROW: 0, NXT: "u", COP: "x", EU: 0},
        "u": {COL: 0, ROW: -1, NXT: "r", COP: "y", EU: 0},
    }
    crt = {"x": x, "y": y}
    for i in range(n * n):
        answer[crt["y"]][crt["x"]] = i + 1
        crt["x"] += combi[direct][COL]
        crt["y"] += combi[direct][ROW]
        if (
            crt[combi[direct][COP]] == combi[direct][EU]
            or answer[crt["y"] + combi[direct][ROW]][crt["x"] + combi[direct][COL]]
        ):
            direct = combi[direct][NXT]
    return answer

테스트 1 〉    통과 (0.09ms, 10.3MB)
테스트 2 〉    통과 (0.04ms, 10.3MB)
테스트 3 〉    통과 (0.30ms, 10.3MB)
테스트 4 〉    통과 (0.11ms, 10.2MB)
테스트 5 〉    통과 (0.16ms, 10.4MB)
테스트 6 〉    통과 (0.03ms, 10.3MB)
테스트 7 〉    통과 (0.42ms, 10.1MB)
테스트 8 〉    통과 (0.52ms, 10.3MB)
테스트 9 〉    통과 (0.24ms, 10.3MB)
테스트 10 〉    통과 (0.21ms, 10.4MB)
테스트 11 〉    통과 (0.31ms, 10.3MB)
테스트 12 〉    통과 (0.41ms, 10.1MB)
테스트 13 〉    통과 (0.00ms, 10.3MB)
테스트 14 〉    통과 (0.01ms, 10.1MB)
테스트 15 〉    통과 (0.56ms, 10.4MB)
테스트 16 〉    통과 (0.09ms, 10.3MB)
테스트 17 〉    통과 (0.44ms, 10.3MB)
테스트 18 〉    통과 (0.20ms, 10.2MB)
테스트 19 〉    통과 (0.27ms, 10.2MB)
테스트 20 〉    통과 (0.47ms, 10.4MB)
테스트 21 〉    통과 (0.21ms, 10.1MB)
테스트 22 〉    통과 (0.05ms, 10.2MB)
테스트 23 〉    통과 (0.01ms, 10.3MB)
테스트 24 〉    통과 (0.27ms, 10.2MB)
테스트 25 〉    통과 (0.66ms, 10.1MB)
테스트 26 〉    통과 (0.07ms, 10.1MB)
테스트 27 〉    통과 (0.11ms, 10.4MB)
테스트 28 〉    통과 (0.54ms, 10.2MB)

2

• 마음 편하게 if문으로 나누는 방법이 훨씬 좋다.

def solution(n):
    if n == 1:
        return [[1]]
    answer = [[0] * n for _ in range(n)]
    y, x = 0, 0
    direct = "r"
    for i in range(n * n):
        answer[y][x] = i + 1
        if direct == "r":
            x += 1
            if x == n - 1 or answer[y][x + 1] != 0:
                direct = "d"
        elif direct == "d":
            y += 1
            if y == n - 1 or answer[y + 1][x] != 0:
                direct = "l"
        elif direct == "l":
            x -= 1
            if x == 0 or answer[y][x - 1] != 0:
                direct = "u"
        elif direct == "u":
            y -= 1
            if y == 0 or answer[y - 1][x] != 0:
                direct = "r"
    return answer

테스트 1 〉    통과 (0.07ms, 10.1MB)
테스트 2 〉    통과 (0.03ms, 10.2MB)
테스트 3 〉    통과 (0.12ms, 10.2MB)
테스트 4 〉    통과 (0.07ms, 10.3MB)
테스트 5 〉    통과 (0.06ms, 10.3MB)
테스트 6 〉    통과 (0.02ms, 10.3MB)
테스트 7 〉    통과 (0.09ms, 10.1MB)
테스트 8 〉    통과 (0.41ms, 10.2MB)
테스트 9 〉    통과 (0.13ms, 10MB)
테스트 10 〉    통과 (0.10ms, 10.2MB)
테스트 11 〉    통과 (0.13ms, 10.1MB)
테스트 12 〉    통과 (0.16ms, 10.2MB)
테스트 13 〉    통과 (0.00ms, 10.3MB)
테스트 14 〉    통과 (0.01ms, 10.4MB)
테스트 15 〉    통과 (0.14ms, 10.2MB)
테스트 16 〉    통과 (0.03ms, 10.2MB)
테스트 17 〉    통과 (0.18ms, 10.2MB)
테스트 18 〉    통과 (0.06ms, 10.1MB)
테스트 19 〉    통과 (0.07ms, 10.1MB)
테스트 20 〉    통과 (0.19ms, 10.3MB)
테스트 21 〉    통과 (0.05ms, 10.2MB)
테스트 22 〉    통과 (0.02ms, 10.4MB)
테스트 23 〉    통과 (0.01ms, 10.2MB)
테스트 24 〉    통과 (0.21ms, 10.3MB)
테스트 25 〉    통과 (0.13ms, 10.1MB)
테스트 26 〉    통과 (0.05ms, 10.2MB)
테스트 27 〉    통과 (0.02ms, 10.2MB)
테스트 28 〉    통과 (0.19ms, 10MB)