LeetCode: Set Matrix Zeroes Solution

1/**
2 * @param {number[][]} matrix
3 * @return {void} Do not return anything, modify matrix in-place instead.
4 */
5var setZeroes = function (matrix) {
6  const N = matrix.length
7  const M = matrix[0].length
8  let setI = new Set()
9  let setJ = new Set()
10
11  for (let i = 0; i < N; i++) {
12    for (let j = 0; j < M; j++) {
13      if (matrix[i][j] === 0) {
14        setI.add(i)
15        setJ.add(j)
16      }
17    }
18  }
19
20  for (let i = 0; i < N; i++) {
21    for (let j = 0; j < M; j++) {
22      if (setI.has(i) || setJ.has(j)) {
23        matrix[i][j] = 0
24      }
25    }
26  }
27
28  return matrix
29}