Skip to content

73、矩阵置零

js
;(function () {
  /**
     * 73. 矩阵置零
     * 
     * 给定一个 m x n 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。
     * 
     * 输入:matrix = [[1,1,1],[1,0,1],[1,1,1]]
     * 输出:[[1,0,1],[0,0,0],[1,0,1]]
     * 
     * 输入:matrix = [[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) 的额外空间,但这仍然不是最好的解决方案。
        你能想出一个仅使用常量空间的解决方案吗?
     * 
     */

  /**
     Do not return anything, modify matrix in-place instead.
    */
  function setZeroes(matrix: number[][]): void {
    // 方法一
    // // 思路:遍历数组中的 0,标记【行中有0】和【列中有0】的项,最后再遍历将这些项全部设置为 0
    // // 备注: 标记的方式可以是 【下标插入数组】或【下标插入对象(为对象属性,值为true)】
    // let rows: number[] = [];
    // let columns: number[] = [];
    // // 标记【行中有0】和【列中有0】的项
    // matrix.map((arr, k) => {
    //     arr.map((item, index) => {
    //         if (item === 0) {
    //             rows.push(k);
    //             columns.push(index);
    //         }
    //     })
    // })
    // let rowsAfterSet = [...Array.from(new Set(rows))];
    // let columnsAfterSet = [...Array.from(new Set(columns))];
    // // 遍历将这些项全部设置为 0
    // matrix.map((arr, k) => {
    //     arr.map((item, index) => {
    //         // 行 / 列 被标记(includes)的项,设置为 0
    //         rowsAfterSet.includes(k) || columnsAfterSet.includes(index) ? matrix[k][index] = 0 : null;
    //     })
    // })

    // 方法二:使用两个标记变量
    // const r = matrix.length, c = matrix[0].length;
    // let flagR0 = false, flagC0 = false;
    // // 1. 判断 第一行 / 第一列 中是否有 0
    // for (let i = 0; i < r; i++) {
    //     matrix[i][0] === 0 ? flagC0 = true : null;
    // }
    // for (let j = 0; j < c; j++) {
    //     matrix[0][j] === 0 ? flagR0 = true : null;
    // }
    // // 2. 其他行/列 处理 第一行/列
    // for (let i = 1; i < r; i++) {
    //     for (let j = 1; j < c; j++) {
    //         if (matrix[i][j] === 0) {
    //             matrix[i][0] = matrix[0][j] = 0;
    //         }
    //     }
    // }
    // // 3. 第一行/列 反过来处理 其他行/列
    // for (let i = 1; i < r; i++) {
    //     for (let j = 1; j < c; j++) {
    //         if (matrix[i][0] === 0 || matrix[0][j] === 0) {
    //             matrix[i][j] = 0;
    //         }
    //     }
    // }
    // // 4. 使用两个标记变量更新第一行与第一列
    // if (flagC0) {
    //     for (let i = 0; i < r; i++) {
    //         matrix[i][0] = 0;
    //     }
    // }
    // if (flagR0) {
    //     for (let j = 0; j < c; j++) {
    //         matrix[0][j] = 0;
    //     }
    // }

    // 方法三:使用一个标记变量
    const m = matrix.length,
      n = matrix[0].length
    let flagCol0 = false
    for (let i = 0; i < m; i++) {
      if (matrix[i][0] === 0) {
        flagCol0 = true
      }
      for (let j = 1; j < n; j++) {
        if (matrix[i][j] === 0) {
          matrix[i][0] = matrix[0][j] = 0
        }
      }
    }
    for (let i = m - 1; i >= 0; i--) {
      for (let j = 1; j < n; j++) {
        if (matrix[i][0] === 0 || matrix[0][j] === 0) {
          matrix[i][j] = 0
        }
      }
      if (flagCol0) {
        matrix[i][0] = 0
      }
    }
  }

  let matrix = [
    [1, 1, 1],
    [1, 0, 1],
    [1, 1, 1]
  ]
  setZeroes(matrix)
  console.log(matrix)
})()