力扣hot100 矩阵置零 标识位

Problem: 73. 矩阵置零

思路

👨‍🏫 参考

复杂度

时间复杂度: $O(nm)$

空间复杂度: $O(1)$

Code

class Solution {public static void setZeroes(int[][] matrix)    {int n = matrix.length;if (n == 0)return;int m = matrix[0].length;int row = 1;//记录第一行是否有 0 int col = 1;//记录第一列是否有 0for (int i = 0; i < m; i++)if (matrix[0][i] == 0)row = 0;for (int i = 0; i < n; i++)if (matrix[i][0] == 0)col = 0;for (int i = 0; i < n; i++)for (int j = 0; j < m; j++){if (matrix[i][j] == 0){// 使用当前行首和列首记录当前行列是否需要置零matrix[i][0] = 0;matrix[0][j] = 0;}}// 根据行首列首 标识位 置0for (int i = 1; i < n; i++)if (matrix[i][0] == 0)Arrays.fill(matrix[i], 0);for (int i = 1; i < m; i++)if (matrix[0][i] == 0)for (int j = 0; j < n; j++)matrix[j][i] = 0;// 最后再处理第一行和第一列if (row == 0)Arrays.fill(matrix[0], 0);if (col == 0){for (int i = 0; i < n; i++) {matrix[i][0] = 0;}}}
}