/**
 * @param {number[][]} matrix
 * @return {void} Do not return anything, modify matrix in-place instead.
 */
var setZeroes2 = function (matrix) {
    let m = matrix.length;
    let n = matrix[0].length;
    let rs = []
    // copy init matrix
    for (let i = 0; i < m; i++) {
        let r = []
        // for(let j=0;j<n;j++) {
        //     r.push(void 0)
        // }
        rs.push(r);
    }

    for (let i = 0; i < m; i++) {
        for (let j = 0; j < n; j++) {
            let v = matrix[i][j]
            if (v === 0) {
                for (let x = 0; x < n; x++) {
                    rs[i][x] = 0;
                }
                for (let y = 0; y < m; y++) {
                    rs[y][j] = 0;
                }
            } else {
                (rs[i][j] === void 0) && (rs[i][j] = v);
            }
        }
    }

    return rs;
};

var setZeroes_ok = function (matrix) {
    let m = matrix.length;
    let n = matrix[0].length;
    let rs = [];
    // init matrix
    for (let i = 0; i < m; i++) {
        rs.push([]);
    }

    for (let i = 0; i < m; i++) {
        for (let j = 0; j < n; j++) {
            let v = matrix[i][j]
            if (v === 0 && rs[i][j] === void 0) {
                for (let x = 0; x < n; x++) {
                    let t = matrix[i][x];
                    if (t !== 0) {
                        rs[i][x] = t;
                    }
                    matrix[i][x] = 0;
                }
                for (let y = 0; y < m; y++) {
                    let t = matrix[y][j];
                    if (t !== 0) {
                        rs[y][j] = t;
                    }
                    matrix[y][j] = 0;
                }
            } else {
                rs[i][j] = v;
            }
        }
    }
};

var setZeroes = function (matrix) {
    let m = matrix.length;
    let n = matrix[0].length;

    let rows = new Set();
    let cols = new Set();

    for (let i = 0; i < m; i++) {
        for (let j = 0; j < n; j++) {
            let v = matrix[i][j]
            if (v === 0) {
                rows.add(i);
                cols.add(j);
            }
        }
    }

    rows.forEach(e => {
        for (let j=0;j<n;j++) {
            matrix[e][j] = 0;
        }
    });

    cols.forEach(e => {
        for (let i=0;i<m;i++) {
            matrix[i][e] = 0;
        }
    });
};
var matrix = [[0, 1, 2, 0], [3, 4, 5, 2], [1, 3, 1, 5]];
setZeroes(matrix);
console.info(matrix);