python 零矩阵

编写一种算法，若M × N矩阵中某个元素为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],
  [0,3,1,0]
]

class Solution:
    def setZeroes(self, matrix: List[List[int]]) -> None:
        """
        解体思路: 两次循环遍历,第一次遍历找到0元素对应的行和列，第二次遍历修改元素
        1.先循环遍历，获取到所有0元素的行索引和列索引
        2.循环遍历行索引和列索引 把行和列都赋值为0 
        """
        # 把出现0元素的行索引添加到row_index的list
        # 把出现0元素的列索引添加到col_index的list
        row_index = []
        col_index = []
        for i, v in enumerate(matrix):
            if 0 in v:
                row_index.append(i)
            for j, val in enumerate(v):
                if 0 == val:
                    col_index.append(j)
        
        # 遍历row_index 和 col_index
        row_len = len(matrix[0])
        for i, v in enumerate(matrix):
            if i in row_index:
                matrix[i] = [0] * row_len  
            for col in col_index:
                v[col] = 0