螺旋矩阵

tech2024-10-08  16

螺旋矩阵

横向遍历m,纵向遍历n-1;横向遍历m-1,纵向遍历n-2;横向遍历m-2,纵向遍历n-3;直到有一方向遍历长度为0时终止。 // time:O(mn) space:O(mn) class Solution: def spiralOrder(self, matrix): if not matrix: return [] cown,cols=len(matrix),len(matrix[0]) judge=1 //1为顺序遍历,-1为逆序遍历 res=[] i,j=0,-1 //初始位置,列为-1,因为代表[0,0]前一个位置 while cown>0 and cols>0: //横向遍历 for _ in range(cols): j+=judge*1 res.append(matrix[i][j] //纵向遍历 for _ in range(cown-1): i+=judge*1 res.append(matrix[i][j]) cown,cols=cown-1,cols-1 judge*=-1 return res

第二种方法:

// time:O(mn) space:O(mn) class Solution: def spiralOrder(self, matrix): if not matrix or not martix[0]: return [] left,right=0,len(matrix[0])-1 top,bottom=0,len(matrix)-1 spiral_path=[] while True: //顶部由左向右 spiral_path.extend(matrix[top][x] for x in range(left,right+1)) top+=1 if top>bottom:break //右部由上到下 spiral_path.extend(matrix[y][right] for y in range(top,bottom+1) right-=1 if left>right:break //底部由右向左 spiral_path.extend(matrix[bottom][x] for x in range(right,left-1,-1)) bottom-=1 if top>bottom:break //左部由下到上 spiral_path.extend(matrix[y][left] for y in range(bottom,top-1)) left+=1 if left>right:break return spiral_path

方法三

// time:O(n) space:O(n) class Solution: def spiralOrder(self, matrix): res=[] if len(matrix)==0:return res rowbegin,rowend,colbegin,colend=0,len(matrix)-1,0,len(matrix[0])-1 while rowbegin<=rowend and colbegin<=colend: //left->right for i in range(colbegin,colend+1): res.append(matrix[rowbegin][i]) rowbegin+=1 //right->bottom for i in range(rowbegin,rowend+1): res.append(matrix[i][colend]) colend-=1 //right->left if rowbegin<=rowend: for i in range(colend,colbegin-1,-1): res.append(matrix[rowend][i]) rowend-=1 //bottom->top if colbegin<=colend: for i in range(rowend,rowbegin-1,-1): res.append(matrix[i][colbegin]) colbegin+=1 return res

螺旋矩阵 II

生成一个 n×n 空矩阵 mat,随后模拟整个向内环绕的填入过程: 定义当前左右上下边界 l,r,t,b,初始值 num = 1,迭代终止值 tar = n * n;当 num <= tar 时,始终按照 从左到右 从上到下 从右到左 从下到上 填入顺序循环,每次填入后: 执行 num += 1:得到下一个需要填入的数字。更新边界:例如从左到右填完后,上边界 t += 1,相当于上边界向内缩 1。 使用num <= tar而不是l < r || t < b作为迭代条件,是为了解决当n为奇数时,矩阵中心数字无法在迭代过程中被填充的问题。 最终返回 mat 即可。 方法一 // time:O(n^2) space:O(n^2) class Solution: def generateMatrix(self, n): left,right,top,bottom=0,n-1,0,n-1 res=[[0 for _ in range(n)] for _ in range(n)] num,target=1,n*n while num<=target: for i in range(left,right+1): res[top][i]=num num+=1 top+=1 for i in range(top,bottom+1): res[i][right]=num num+=1 right-=1 for i in range(right,left-1,-1): res[bottom][i]=num num+=1 bottom-=1 for i in range(bottom,top-1,-1): res[i][left]=num num+=1 left+=1 return res
最新回复(0)