[5/9] 54. Spiral Matrix

class Solution:
    def spiralOrder(self, matrix: List[List[int]]) -> List[int]:
        spiral = []
        m, n = len(matrix), len(matrix[0])
        total = m * n
        directions = [(0,1), (1,0), (0, -1), (-1, 0)]
        cur_directions = 0
        cur = (0,0)
        curlen = 0
        while curlen < total:
            spiral.append(matrix[cur[0]][cur[1]])
            curlen += 1
            matrix[cur[0]][cur[1]] = -101
            next_pos = (cur[0] + directions[cur_directions][0], cur[1] + directions[cur_directions][1])
            if next_pos[0] < 0 or next_pos[1] < 0 or next_pos[0] == m or next_pos[1] == n or -101 ==  matrix[next_pos[0]][next_pos[1]]:
                cur_directions = (cur_directions + 1) % 4
            cur = (cur[0] + directions[cur_directions][0], cur[1] + directions[cur_directions][1])
        return spiral

댓글 다실때 오른쪽 아래에 보시면 </> 아이콘이 있는데 이걸로 코드 스니펫 입력 가능해요