前言
遇到一个输出螺旋矩阵(回型矩阵)的题目,这个题目本科的时候就做过多次了,但是最近发现一个用Python zip函数巧妙转置矩阵的方法,记录一下。
题解
题目:输入一个矩阵,按照从外向里以顺时针的顺序依次打印出每一个数字。
这个题只需要按要求模拟过程即可,Java代码如下:
class Solution { public int[] spiralOrder(int[][] matrix) { if(matrix.length == 0) return new int[0]; int l = 0, r = matrix[0].length - 1, t = 0, b = matrix.length - 1, x = 0; int[] res = new int[(r + 1) * (b + 1)]; while(true) { for(int i = l; i <= r; i++) res[x++] = matrix[t][i]; // left to right. if(++t > b) break; for(int i = t; i <= b; i++) res[x++] = matrix[i][r]; // top to bottom. if(l > --r) break; for(int i = r; i >= l; i--) res[x++] = matrix[b][i]; // right to left. if(t > --b) break; for(int i = b; i >= t; i--) res[x++] = matrix[i][l]; // bottom to top. if(++l > r) break; } return res; }}
有一个Python版本的,巧用zip函数的特性将矩阵转置,代码非常简洁,Python代码如下:
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