零矩阵

---

1. 题目描述：

编写一种算法，若 M × N 矩阵中某个元素为 0，则将其所在的行与列清零。

2. 力扣官方题解
3. ChatGPT

遍历整个矩阵，记录所有为 0 的元素的行号和列号。
根据记录的行号和列号，将对应的行与列清零。

function setZeroes(matrix) {
  const m = matrix.length; // 矩阵的行数
  const n = matrix[0].length; // 矩阵的列数

  const zeroRows = new Set(); // 记录为0的行号
  const zeroCols = new Set(); // 记录为0的列号

  // 遍历矩阵，记录为0的元素的行号和列号
  for (let i = 0; i < m; i++) {
    for (let j = 0; j < n; j++) {
      if (matrix[i][j] === 0) {
        zeroRows.add(i);
        zeroCols.add(j);
      }
    }
  }

  // 根据记录的行号和列号，将对应的行与列清零
  for (let i = 0; i < m; i++) {
    for (let j = 0; j < n; j++) {
      if (zeroRows.has(i) || zeroCols.has(j)) {
        matrix[i][j] = 0;
      }
    }
  }
}

// 示例
const matrix = [
  [1, 1, 1],
  [1, 0, 1],
  [1, 1, 1],
];

setZeroes(matrix);
console.log(matrix);

1. 我的解法

var setZeroes = function (matrix) {
  // 记录哪些行包含 0
  let row = [];
  // 记录哪些列包含 0
  let column = [];
  for (let i = 0; i < matrix.length; i++) {
    for (let j = 0; j < matrix[i].length; j++) {
      if (0 === matrix[i][j]) {
        if (!row.includes(i)) {
          row.push(i);
        }
        if (!column.includes(j)) {
          column.push(j);
        }
      }
    }
  }

  if (row.length === 0 && column.length === 0) {
    return matrix;
  } else {
    // 将所有 0 所在的行清零
    row.forEach((row) => {
      for (let i = 0; i < matrix[row].length; i++) {
        matrix[row][i] = 0;
      }
    });
    // 将所有 0 所在的列清零
    column.forEach((column) => {
      for (let i = 0; i < matrix.length; i++) {
        matrix[i][column] = 0;
      }
    });
  }
};

const matrix1 = [
  [1, 1, 1],
  [1, 0, 1],
  [1, 1, 1],
];
setZeroes(matrix1);
console.log(matrix1); // 输出 [[7, 4, 1], [8, 5, 2], [9, 6, 3]]