Leetcode 73 矩陣置零(python)

2021-08-25 16:42:13 字數 1504 閱讀 2414

方法一:使用兩個list分別記錄行列上需要置零的行列的index,最差情況的空間複雜度為o(m+n)

class

solution:

defsetzeroes

(self,matrix):

rows=len(matrix)

cols=len(matrix[0])

row_index=

col_index=

for row in range(rows):

for col in range(cols):

if matrix[row][col]==0:

for row in row_index:

matrix[row]=[0]*cols

for col in col_index:

for row in range(rows):

matrix[row][col]=0

方法二(常數空間複雜度):把行列置零記錄的list的功能,直接放到原來矩陣的第一行和第一列,在遍歷元素的時候把需要置零的行在第一列對應位置置零,需要置零的列在第一行對應位置置零,後續再根據第一行和第一列的標記補零。

這裡存在的問題是,第一行和第一列本來存在的零和後來補上的零會混淆,原來的零具備乙個功能是把第一行和第一列置零,所以在一開始的時候,需要遍歷第一行和第一列,確定是否有零。然後在最後對第一行和第一列進行置零操作。

class

solution:

defsetzeroes

(self,matrix):

rows=len(matrix)

cols=len(matrix[0])

row_zero=false

col_zero=false

for i in matrix[0]:

if i==0:

row_zero=true

for j in range(rows):

if matrix[j][0]==0:

col_zero=true

for row in range(1,rows):

for col in range(1,cols):

if matrix[row][col]==0:

matrix[row][0]=0

matrix[0][col]=0

for row in range(1,rows):

if matrix[row][0]==0:

matrix[row]=[0]*cols

for col in range(1,cols):

if matrix[0][col]==0:

for row in range(1,rows):

matrix[row][col]=0

if row_zero:

matrix[0]=[0]*cols

if col_zero:

for row in range(rows):

matrix[row][0]=0

leetcode 73 矩陣置零

給定乙個 m x n 的矩陣,如果乙個元素為 0,則將其所在行和列的所有元素都設為 0。請使用原地演算法。示例 1 輸入 1,1,1 1,0,1 1,1,1 輸出 1,0,1 0,0,0 1,0,1 示例 2 輸入 0,1,2,0 3,4,5,2 1,3,1,5 輸出 0,0,0,0 0,4,5,0...

leetcode73 矩陣置零

給定乙個 m x n 的矩陣,如果乙個元素為 0,則將其所在行和列的所有元素都設為 0。請使用原地演算法。示例 1 輸入 1,1,1 1,0,1 1,1,1 輸出 1,0,1 0,0,0 1,0,1 空間複雜度為o mn 的解法自不用多說,直接新建乙個和matrix等大小的矩陣,然後一行一行的掃,只...

leetcode 73 矩陣置零

給定乙個 m x n 的矩陣,如果乙個元素為 0,則將其所在行和列的所有元素都設為 0。請使用原地演算法。示例 1 輸入 1,1,1 1,0,1 1,1,1 輸出 1,0,1 0,0,0 1,0,1 示例 2 輸入 0,1,2,0 3,4,5,2 1,3,1,5 輸出 0,0,0,0 0,4,5,0...