问题: 给定一个 m x n 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法 。 示例 1: 输入:matrix = [[1,1,1],[1,0,1],[1,1,1]] 输出:[[1,0,1],[0,0,0],[1,0,1]] 示例 2: 输入: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]] 回答: function setZeroes(matrix) { const m = matrix.length; const n = matrix[0].length; let firstRowHasZero = false; let firstColHasZero = false; // 检查第一行是否有 0 for (let j = 0; j < n; j++) { if (matrix[0][j] === 0) { firstRowHasZero = true; break; } } // 检查第一列是否有 0 for (let i = 0; i < m; i++) { if (matrix[i][0] === 0) { firstColHasZero = true; break; } } // 用第一行和第一列做标记 for (let i = 1; i < m; i++) { for (let j = 1; j < n; j++) { if (matrix[i][j] === 0) { matrix[i][0] = 0; matrix[0][j] = 0; } } } // 根据标记置零(不包含第一行、第一列) for (let i = 1; i < m; i++) { for (let j = 1; j < n; j++) { if (matrix[i][0] === 0 || matrix[0][j] === 0) { matrix[i][j] = 0; } } } // 处理第一行 if (firstRowHasZero) { for (let j = 0; j < n; j++) { matrix[0][j] = 0; } } // 处理第一列 if (firstColHasZero) { for (let i = 0; i < m; i++) { matrix[i][0] = 0; } } }