原题链接:73. 矩阵置零
解题思路:
- 遍历矩阵,找到所有为
0
的坐标,将它们都存入栈。 - 从栈中依次弹出坐标,并将对应的行、列元素都设置为0。
/**
* @param {number[][]} matrix
* @return {void} Do not return anything, modify matrix in-place instead.
*/
var setZeroes = function (matrix) {
let stack = []; // 使用栈存储矩阵中为0的坐标
let m = matrix.length; // 缓存矩阵的行数
let n = matrix[0].length; // 缓存矩阵的列数
// 遍历矩阵的每个坐标
for (let i = 0; i < m; i++) {
for (let j = 0; j < n; j++) {
// 如果当前位置为0,将坐标存入栈
if (matrix[i][j] === 0) {
stack.push([i, j]);
}
}
}
// 将所有坐标的行和列都设为0
while (stack.length) {
// 每次取出一个坐标
const [x, y] = stack.pop();
// 将当前坐标所在的列设为0
for (let i = 0; i < m; i++) {
matrix[i][y] = 0;
}
// 将当前坐标所在的行设为0
for (let i = 0; i < n; i++) {
matrix[x][i] = 0;
}
}
};