LeetCode——行列のゼロを設定します

Leetcode Set Matrix Zeroes

m x n行列が与えられ、要素が0の場合、その行と列全体を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]
  ]
• ファローアップ：
  O（mn）スペースを使用した簡単な解決策はおそらく悪い考えです。
  単純な改善ではO（m + n）スペースを使用しますが、それでも最善の解決策ではありません。
  一定の空間ソリューションを考案できますか？

class Solution: def setZeroes(self, matrix: List[List[int]]) -> None: ''' Do not return anything, modify matrix in-place instead. ''' if not matrix: return n,m,k=len(matrix),len(matrix[0]),0 while k<m and matrix[0][k]: k=k+1 for i in range(1,n): for j in range(m): if not matrix[i][j]: matrix[i][0]=matrix[0][j]=0 for i in range(1,n): for j in range(m-1,-1,-1): if not matrix[i][0] or not matrix[0][j]: matrix[i][j]=0 if k<m: matrix[0]=[0]*m