1.1矩阵置零
给定一个 m x n 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0
。请使用原地算法
。
输入
:matrix = [[0,1,2,0],[3,4,5,2],[1,3,1,5]]
输出
:[[0,0,0,0],[0,4,5,0],[0,3,1,0]]
原地算法(in-place algorithm)
基本上不需要额外辅助的数据结构,然而,允许少量额外的辅助变量来转换数据的算法
。当算法运行时,输入的数据通常会被要输出的部分覆盖掉。不是原地算法有时候称为非原地(not-in-place)或不得其所(out-of-place)。
代码
def setZeroes(self,matrix):m,n=len(matrix),len(matrix[0])row,col=[False]*m,[False]*nfor i in range(m):for j in range(n):if matrix[i][j]==0:row[i]=col[j]=True #把0所在行和列标记为Truefor i in range(m):for j in range(n):if row[i] or col[j]:matrix[i][j]=0return matrix
1.2螺旋矩阵
给你一个 m 行 n 列的矩阵 matrix ,请按照 顺时针螺旋顺序
返回矩阵中的所有元素。
输入
:matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出
:[1,2,3,6,9,8,7,4,5]
代码
def spiralOrder(self,matrix):m,n=len(matrix),len(matrix[0])upper,left,right,down=0,0,n-1,m-1 #四个位置边界的下标res=[]while True:for i in range(left,right+1): #向右移动到最右res.append(matrix[upper][i])upper+=1 #缩小上边界,去掉已经遍历过的行if upper>down:breakfor i in range(upper,down+1): #向下移动到最底部res.append(matrix[i][right])right-=1 #缩小右边界,去掉已经遍历过的列if right<left:breakfor i in range(right,left-1,-1): #往左res.append(matrix[down][i])down-=1 #缩小下边界,去掉遍历过的行if down<upper:breakfor i in range(down,upper-1,-1): #向上res.append(matrix[i][left])left+=1 #缩小左边界,去掉已经遍历过的列if left>right:breakreturn res
python的range函数基础用法
#range(start, stop, step)=range(初值, 终值, 步长)
#取值范围是[start,stop)
#range()函数只适用于整数,步长可以为正值也可以为负值for i in range(4): print(i)#0 1 2 3for i in range(1,4):print(i) #1,2,3for i in range(1,15,4):print(i) #1,5,9,13for i in range(4,1,-1):print(i) #4,3,2n=list(range(4))
print(n) #[0,1,2,3]
参考代码