python矩阵置零(leetcode)

给定一个m x n 的矩阵, 如果一个元素为0，则将其所在的行和列所有元素都设为0，请使用原地算法。

思路：使用一个列表记录等于零的坐标，然后进行替换

class Solution01:
    def setZeroes(self, matrix):
        tmp = []
        for i in range(len(matrix)):
            for j in range(len(matrix[0])):
                if matrix[i][j] == 0:
                    tmp.append((i,j))
        for cow,col in tmp:
            matrix[cow] = [0]*len(matrix[0])
            for i in range(len(matrix)):
                matrix[i][col] = 0
        return matrix