18.矩阵置零(原地算法)

var setZeroes = function (matrix) {const rows = matrix.length;const cols = matrix[0].length;// 标记第一行和第一列是否本身包含0(用于最后处理首行首列)let firstRowHasZero = false;let firstColHasZero = false;// 第一步:检查第一行是否有0for (let j = 0; j < cols; j++) {if (matrix[0][j] === 0) {firstRowHasZero = true;break;}}// 第二步:检查第一列是否有0for (let i = 0; i < rows; i++) {if (matrix[i][0] === 0) {firstColHasZero = true;break;}}// 第三步:用第一行和第一列标记其他行和列是否需要置零for (let i = 1; i < rows; i++) { // 从第1行开始(跳过首行)for (let j = 1; j < cols; j++) { // 从第1列开始(跳过首列)if (matrix[i][j] === 0) {// 用首行对应列标记该列需要置零matrix[0][j] = 0;// 用首列对应行标记该行列需要置零matrix[i][0] = 0;}}}// 第四步:根据首行首列的标记,将其他行和列置零for (let i = 1; i < rows; i++) {for (let j = 1; j < cols; j++) {// 如果当前行的首列是0,或当前列的首行是0,则置零if (matrix[i][0] === 0 || matrix[0][j] === 0) {matrix[i][j] = 0;}}}// 第五步:处理第一行(如果本身有0,整行置零)if (firstRowHasZero) {for (let j = 0; j < cols; j++) {matrix[0][j] = 0;}}// 第六步:处理第一列(如果本身有0,整列置零)if (firstColHasZero) {for (let i = 0; i < rows; i++) {matrix[i][0] = 0;}}
};解题思路
将第一行和第一列拿出来做标记,然后遍历剩余的矩阵,如果该行或该列的第一个标记为0,就把该值赋值为0,空间复杂度为O(1)
详细解法
1.检查第一行和第一列是否有0
2.用第一行和第一列标记其他行和列是否需要置零,如果该元素为0,就标记该行和该列第一个元素为0
3.根据首行首列为0的元素将其他元素也变为0
4.处理首行和首列,如果原本就有0,将整行或整列变为0
