LeetCode-面試題29、順時針打印矩陣-簡單

LeetCode-面試題29、順時針打印矩陣-簡單

輸入一個矩陣，按照從外向裏以順時針的順序依次打印出每一個數字。

示例 1：

輸入：matrix = [[1,2,3],[4,5,6],[7,8,9]]
輸出：[1,2,3,6,9,8,7,4,5]

示例 2：

輸入：matrix = [[1,2,3,4],[5,6,7,8],[9,10,11,12]]
輸出：[1,2,3,4,8,12,11,10,9,5,6,7]

限制：

• 0 <= matrix.length <= 100
• 0 <= matrix[i].length <= 100

注意：本題與主站 54 題相同：https://leetcode-cn.com/problems/spiral-matrix/

 

代碼：

class Solution:
    def spiralOrder(self, matrix: List[List[int]]) -> List[int]:
        ans = []
        while matrix:
            ans += matrix.pop(0)
            l = len(matrix)
            for i in range(l):
                matrix[i] = matrix[i][::-1]
            if l == 1:
                ans += matrix[0]
                break
            matrix = list(zip(*matrix))
        return ans

# class Solution:
#     def spiralOrder(self, matrix: List[List[int]]) -> List[int]:
#         return list(matrix[0]) + self.spiralOrder(list(zip(*matrix[1:]))[::-1]) if matrix else []