Coding Trainer

Set Matrix Zeroes

MediumMatrix Traversalk-matrixLC #73

Problem

Set Matrix Zeroes

Given an m × n integer matrix, if an element is 0, set its entire row and column to 0. Do it in-place.

Example 1:

Input:
matrix = [
  [1,1,1],
  [1,0,1],
  [1,1,1]
]
Output:
[
  [1,0,1],
  [0,0,0],
  [1,0,1]
]

Example 2:

Input:
matrix = [
  [0,1,2,0],
  [3,4,5,2],
  [1,3,1,5]
]
Output:
[
  [0,0,0,0],
  [0,4,5,0],
  [0,3,1,0]
]

Constraints:

  • m == matrix.length, n == matrix[0].length
  • 1 ≤ m, n ≤ 200
  • -2³¹ ≤ matrix[i][j] ≤ 2³¹ - 1

Follow-up: Could you devise an algorithm that uses only O(1) extra space?