54. 螺旋矩阵
力扣上已经有很多题解了,但是总觉得很多题解有点复杂不太符合直觉,下面两种解法是我认为比较简洁易懂的。(虽然这两种方法的用时与内存都不算特别优化。
解法一:根据螺旋规律输出这种方法有C++版本,这个思路很符合问题的常规思考方向,又可以比较简洁地遍历四个方向。
观察螺旋规律
从 matrix[0][-1] 为起点,开始(这里的起点不是指实际上的第一排第一列的元素,而只是一个位置控制。如果是从matrix[0][0]开始,则在顺时针的方案中第一个 *** 作的元素会是matrix[0][1],最后会溢出)向右出发,那么第一次会走 n 步,n为列数向下转,会走 m - 1 步,m为行数向左转,n - 1 步向上转,m - 2 步从上可以看出,n-->m-1-->n-1-->m-2-->n-2-->m-3-->...-->1,步数在行与列依次减少一。因此,利用数组sz的值控制好步数和步数的变化,便可将所有元素依次填入ans数组。
其中dd为方向数组,分别为向右(纵坐标+1),向下(横坐标+1),向左(纵坐标-1)与向上(横坐标-1)。通过d = (d + 1) % len(dd)遍历方向数组即可顺序切换方向。
如果是逆时针,则方向数组顺序为:向下(横坐标+1),向右(纵坐标+1),向上(横坐标-1)与向左(纵坐标-1)
个人编写的python版本:
class Solution: def spiralOrder(self, matrix: List[List[int]]) -> List[int]: sz = [len(matrix[0]), len(matrix)] ans = [0] * sz[0] * sz[1] sz[1] = sz[1] - 1 dd = [[0, 1], [1, 0], [0, -1], [-1, 0]] IDx = 0 d = 0 x = 0 y = -1 while IDx != len(ans): for i in range(sz[d % 2]): x = x + dd[d][0] y = y + dd[d][1] ans[IDx] = matrix[x][y] IDx = IDx + 1 sz[d % 2] = sz[d % 2] - 1 d = (d + 1) % len(dd) return ans
如果是逆时针螺旋,则需要对应修改数组sz、方向数组dd与起始位置xy:
sz[0] = sz[0] - 1 dd = [[1, 0], [0, 1], [-1, 0], [0, -1]] IDx = 0 d = 0 x = -1 y = 0
解法二:削苹果这个解法相当简洁,思路跟削苹果一样,旋转一下矩阵,然后削去最上方的头,直到整颗苹果削完。
class Solution: def spiralOrder(self, matrix: List[List[int]]) -> List[int]: res = [] while matrix: # 削头(第一层) res += matrix.pop(0) # 将剩下的逆时针转九十度,等待下次被削 matrix = List(zip(*matrix))[::-1] return res
其中:
List pop() 默认是pop最后一个元素,默认为 index=-1,这里pop的是第一行(List matrix的第一个元素)。(这 *** 作很符合直觉,一看到这个问题,首先就很容易想到用栈这个数据结构来解决。zip() 是python的内置函数,在python3中返回的是一个对象,需要手动List()转换。 总结以上是内存溢出为你收集整理的【python】54. 螺旋矩阵全部内容,希望文章能够帮你解决【python】54. 螺旋矩阵所遇到的程序开发问题。
如果觉得内存溢出网站内容还不错,欢迎将内存溢出网站推荐给程序员好友。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)