Leetcode 54. 螺旋矩阵(Python3)

54. 螺旋矩阵

给定一个包含 m x n 个元素的矩阵（m 行, n 列），请按照顺时针螺旋顺序，返回矩阵中的所有元素。

示例 1:

输入:
[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]
输出: [1,2,3,6,9,8,7,4,5]

示例 2:

输入:
[
  [1, 2, 3, 4],
  [5, 6, 7, 8],
  [9,10,11,12]
]
输出: [1,2,3,4,8,12,11,10,9,5,6,7]

---

class Solution:
    def spiralOrder(self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: List[int]
        """
        if not matrix:return []
        res = []
        m,n,c,j = len(matrix),len(matrix[0]),0,0
        while c < m*n:
            for i in range(j,n-j):
                res.append(matrix[j][i])
                c += 1
            for i in range(j+1,m-j):
                res.append(matrix[i][n-j-1])
                c += 1
            for i in range(n-j-2,j-1,-1):
                res.append(matrix[m-j-1][i])
                c += 1
            for i in range(m-j-2,j,-1):
                res.append(matrix[i][j])
                c += 1
            j += 1
        return res[:m*n]

---

DFS

class Solution(object):
    def spiralOrder(self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: List[int]
        """
        # 取首行，去除首行后，对矩阵翻转来创建新的矩阵，
        # 再递归直到新矩阵为[],退出并将取到的数据返回
        ret = []
        if matrix == []:
            return ret
        ret.extend(matrix[0]) # 上侧
        new = [reversed(i) for i in matrix[1:]]
        if new == []:
            return ret
        r = self.spiralOrder([i for i in zip(*new)])
        ret.extend(r)
        return ret