Python/Spiral Array

원문

문제는 다음과 같습니다:

6 6

  0   1   2   3   4   5
 19  20  21  22  23   6
 18  31  32  33  24   7
 17  30  35  34  25   8
 16  29  28  27  26   9
 15  14  13  12  11  10

위처럼 6 6이라는 입력을 주면 6 X 6 매트릭스에 나선형 회전을 한 값을 출력해야 합니다.

풀이

#!/usr/bin/python
# -*- coding: utf-8 -*-

def fill_round(matrix, num, mx, my, base_x, base_y):
    ix = base_x
    iy = base_y

    # LR 진행
    for n in range(0, mx):
        matrix[iy][ix] = num
        num = num + 1
        ix = ix + 1
    ix = ix - 1
    mx = mx - 1

    # UD 진행
    iy = iy + 1
    for n in range(1, my):
        matrix[iy][ix] = num
        num = num + 1
        iy = iy + 1
    iy = iy - 1
    my = my - 1

    if my < 1:
        return

    # RL 진행
    ix = ix - 1
    for n in range(mx - 1, -1, -1):
        matrix[iy][ix] = num
        num = num + 1
        ix = ix - 1
    ix = ix + 1
    mx = mx - 1

    if mx < 1:
        return

    # DU 진행
    iy = iy - 1
    for n in range(my - 1, 0, -1):
        matrix[iy][ix] = num
        num = num + 1
        iy = iy - 1
    return num

def solution(x, y):
    if x < 0 or y < 0:
        return [[]]
    matrix = [[-1 for i in range(0, x)] for j in range(0, y)]

    num = 0
    w = x
    h = y
    i = 0
    while True:
        num = fill_round(matrix, num, w, h, i, i)
        i = i + 1
        w = w - 2
        h = h - 2
        if w <= 0 or h <= 0:
            break
    return matrix

matrix = solution(6, 6)
print '\n'.join([' '.join(['%3u' % n for n in m]) for m in matrix])

print ''

matrix = solution(9, 7)
print '\n'.join([' '.join(['%3d' % n for n in m]) for m in matrix])

결과

  0   1   2   3   4   5
 19  20  21  22  23   6
 18  31  32  33  24   7
 17  30  35  34  25   8
 16  29  28  27  26   9
 15  14  13  12  11  10

  0   1   2   3   4   5   6   7   8
 27  28  29  30  31  32  33  34   9
 26  47  48  49  50  51  52  35  10
 25  46  59  60  61  62  53  36  11
 24  45  58  57  56  55  54  37  12
 23  44  43  42  41  40  39  38  13
 22  21  20  19  18  17  16  15  14