LeetCode刷題——Matrix zeroing矩陣置零

矩陣置零題目的思路探討與源碼
矩陣置零的題目如下圖，核心把矩陣中當前0元素所在的行和列都重置爲0，並且只能在當前矩陣上改動，不能新建矩陣，所以要使用下標進行標記，並最終重新賦值。


本人在看到該題目後，認爲只需要記錄下所有0元素所在的行和列即可，然後直接遍歷進行賦值。但是實際上這樣的代碼不是最優的，因爲進行了兩次雙重for循環，代碼如下所示：

#噴火龍與水箭龜
import numpy as np
matrix=[[0,1,2,0],  [3,4,5,2],  [1,3,1,5]]
mS=np.shape(matrix)
mRow=mS[0]
mCol=mS[1]
setRow=set()
setCol=set()
for zi in range(mRow):
	for zj in range(mCol):
		if(matrix[zi][zj]==0):
			setRow.add(zi)
			setCol.add(zj)
listRow=list(setRow)
listCol=list(setCol)
for zi in range(mRow):
	for zj in range(mCol):
		if(zj in listCol or zi in listRow):
			matrix[zi][zj]=0
print(matrix)

在上述代碼中，zi和zj是兩個移動的下標，zi代表行下標、zj代表列下標。首先通過遍歷找到所有元素值爲0的行和列，然後進行記錄到集合，這樣不會重複。然後轉爲list進行保存，再使用雙重for循環對行爲0和列爲0的列表進行遍歷，對矩陣進行賦值，最終得到結果。
最終輸入是[[0,1,2,0], [3,4,5,2], [1,3,1,5]]，得到結果[[0, 0, 0, 0], [0, 4, 5, 0], [0, 3, 1, 0]] ，是正確的。需要注意的是，本代碼是作者的測試代碼，如需放到LeetCode上運行，需要進行函數縮進和略微修改纔可以運行。
最終結果如下：

從結果來說，本人的思路顯然不太好，效率一般。或許結合原地算法或矩陣乘法等可以提升，希望朋友們能夠多多指教。