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上运行，需要进行函数缩进和略微修改才可以运行。
最终结果如下：

从结果来说，本人的思路显然不太好，效率一般。或许结合原地算法或矩阵乘法等可以提升，希望朋友们能够多多指教。