面试题 01.08. 零矩阵

面试题 01.08. 零矩阵

https://leetcode-cn.com/problems/zero-matrix-lcci/submissions/

难度中等13收藏分享切换为英文关注反馈

编写一种算法，若M × N矩阵中某个元素为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]
]

执行用时：140 ms, 在所有 JavaScript 提交中击败了11.78%的用户

内存消耗：39.7 MB, 在所有 JavaScript 提交中击败了91.03%的用户

/**
 * @param {number[][]} matrix
 * @return {void} Do not return anything, modify matrix in-place instead.
 */
var setZeroes = function (matrix) {
    var arr = new Array();
    for (var i = 0; i < matrix.length; i++)  //有matrix.length行  有matrix[0].length列
    {
        for (var j = 0; j < matrix[0].length; j++) {
            if (matrix[i][j] == 0) arr.push(i, j);  //记录了为0的行与列
        }
    }
    for (var i = 0; i < arr.length; i++) {
        if (i % 2 == 0) {   //消除行
            for (var j = 0; j < matrix[0].length; j++) {
                matrix[arr[i]][j] = 0;
            }
        }
        else {   //消除列
            for (var j = 0; j < matrix.length; j++) {
                matrix[j][arr[i]] = 0;
            }
        }

    }
};