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]*n
for i in range(m):
for j in range(n):
if matrix[i][j]==0:
row[i]=col[j]=True #把0所在行和列标记为True
for i in range(m):
for j in range(n):
if row[i] or col[j]:
matrix[i][j]=0
return 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:
break
for i in range(upper,down+1): #向下移动到最底部
res.append(matrix[i][right])
right-=1 #缩小右边界,去掉已经遍历过的列
if right<left:
break
for i in range(right,left-1,-1): #往左
res.append(matrix[down][i])
down-=1 #缩小下边界,去掉遍历过的行
if down<upper:
break
for i in range(down,upper-1,-1): #向上
res.append(matrix[i][left])
left+=1 #缩小左边界,去掉已经遍历过的列
if left>right:
break
return res
python的range函数基础用法
#range(start, stop, step)=range(初值, 终值, 步长)
#取值范围是[start,stop)
#range()函数只适用于整数,步长可以为正值也可以为负值
for i in range(4):
print(i)#0 1 2 3
for i in range(1,4):
print(i) #1,2,3
for i in range(1,15,4):
print(i) #1,5,9,13
for i in range(4,1,-1):
print(i) #4,3,2
n=list(range(4))
print(n) #[0,1,2,3]