编辑代码

class Solution:
    def setZeros(self,matrix):
        row = len(matrix)
        col = len(matrix[0])
        row_Flag = False
        col_Flag = False
        for j in range(col):
            if matrix[0][j] == 0:
                row_Flag = True
                break

        for i in range(row):
            if matrix[i][0] == 0:
                col_Flag = True
                break
        for i in range(0,row):
            for j in range(0,col):
                if matrix[i][j] == 0:
                    matrix[i][0] = matrix[0][j] = 0
        print(matrix)
        for i in range(0,row):
            for j in range(0,col):
                if matrix[i][0] == 0 or matrix[0][j] == 0:
                    matrix[i][j] = 0
        
        if row_Flag:
            for j in range(col):
                matrix[0][j] = 0
        if col_Flag:
            for i in range(row):
                matrix[i][0] = 0

if __name__ == '__main__':
    # matrix = [[5,0,2,5],[3,0,0,2],[1,3,1,5]]
    matrix = [[0, 1, 2, 0], [3, 4, 5, 2], [1, 3, 1, 5]]
    Solution().setZeros(matrix)
    print(matrix)