leetcode54／剑指offer29.螺旋矩阵（字节跳动面试题、VIVO面试题）、leetcode59.螺旋矩阵 II

题目一：leetcode54.螺旋矩阵

1.题目描述

给定一个包含 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]

2.解题思路

首先弹出矩阵第一行，然后想办法弹出矩阵的最后一列。可以用zip(*matrix)[::-1]实现。

zip()是Python的一个内建函数，它接受一系列可迭代的对象作为参数，将对象中对应的元素打包成一个个tuple（元组），第0个元组对应于所有参数的第0个元素，第1个元组对应于所有参数的第1个元素，依此类推，然后返回由这些tuples组成的list（列表）。若传入参数的长度不等，则返回list的长度和参数中长度最短的对象相同。
zip([1,2,3,4],[5,6,7,8])会返回[(1, 5), (2, 6), (3, 7), (4, 8)]
但是如果输入的参数在一个列表当中，比如l =[[1,2,3,4],[5,6,7,8]]，就需要进行解包裹，就是在传递列表时，让列表的每一个元素对应一个位置参数。在调用zip时使用*，是为了提醒Python：我想要把args拆成分散的多个个元素，分别传递给多个位置参数。
l = [[1,2,3,4],[5,6,7,8]]
zip(*l)所实现的功能与上面的代码相同，只不过输入的参数由列表解包得到。

---

解法二：（VIVO）

绘制螺旋轨迹路径，我们发现当路径超出界限或者进入之前访问过的单元格时，会顺时针旋转方向。

起点就是（0，0）位置，但是方向数组一定要注意，不能随便写，开始时是要往右走，到了边界或者访问过的位置后，就往下，然后往左，再往上，所以 dirs 数组的顺序是 右->下->左->上

3.代码实现

class Solution(object):
    def spiralOrder(self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: List[int]
        """
        res = []
        while matrix:
            res += matrix.pop(0)
            matrix=zip(*matrix)[::-1]
        return res

解法二：

class Solution(object):
    def spiralOrder(self, matrix):
        if not matrix: return []
        R, C = len(matrix), len(matrix[0])
        seen = [[False] * C for _ in matrix]
        ans = []
        dr = [0, 1, 0, -1]
        dc = [1, 0, -1, 0]
        r = c = di = 0
        for _ in range(R * C):
            ans.append(matrix[r][c])
            seen[r][c] = True
            cr, cc = r + dr[di], c + dc[di]
            if 0 <= cr < R and 0 <= cc < C and not seen[cr][cc]:
                r, c = cr, cc
            else:
                di = (di + 1) % 4
                r, c = r + dr[di], c + dc[di]
        return ans

题目二：leetcode59.螺旋矩阵 II

1.题目描述

给定一个正整数 n，生成一个包含 1 到 n2 所有元素，且元素按顺时针顺序螺旋排列的正方形矩阵。

示例:

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

2.解题思路

参考链接：https://leetcode-cn.com/problems/spiral-matrix-ii/solution/spiral-matrix-ii-mo-ni-fa-she-ding-bian-jie-qing-x/

3.代码实现

class Solution(object):
    def generateMatrix(self, n):
        """
        :type n: int
        :rtype: List[List[int]]
        """
        mat = [[0 for _ in range(n)] for _ in range(n)]
        num, tar = 1, n * n
        l,t,b,r = 0,0, n-1,n-1
        while num <= tar:
            for i in range(l,r+1):
                mat[t][i] = num
                num += 1
            t+=1
            for i in range(t,b+1):
                mat[i][r] = num
                num += 1
            r-=1
            for i in range(r,l-1,-1):
                mat[b][i] = num
                num += 1
            b-=1
            for i in range(b,t-1,-1):
                mat[i][l] = num
                num += 1
            l+=1
        return mat