leetcode 矩阵置零（js实现）

矩阵置零

给定一个 m x n 的矩阵，如果一个元素为 0，则将其所在行和列的所有元素都设为 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] ]

解题思路

利用额外空间进行解题，首先双层循环遍历，记录下元素为0的位置i，j，分别存储到数组r和l中，再次双层循环遍历，判断数组r,l中存在那些数与i和j相同，相同则将相应元素置0.

/**
 * @param {number[][]} matrix
 * @return {void} Do not return anything, modify matrix in-place instead.
 */
var setZeroes = function(matrix) {
        var r = []
        var l = []
        for(var i=0;i<matrix.length;i++){
            for(var j=0;j<matrix[0].length;j++){
                if(matrix[i][j] === 0){
                    r.push(i)
                    l.push(j)
                }
            }
        }
        for(var i=0;i<matrix.length;i++){
            for(var j=0;j<matrix[0].length;j++){
                if( r.indexOf(i) != -1 || l.indexOf(j) != -1){
                    matrix[i][j] = 0
                }
            }
        }
    return matrix
};