1.题目描述
给定一个
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]]提示:
m == matrix.length
n == matrix[0].length
1 <= m, n <= 200
-231 <= matrix[i][j] <= 231 - 1
进阶:
- 一个直观的解决方案是使用
O(mn)
的额外空间,但这并不是一个好的解决方案。- 一个简单的改进方案是使用
O(m + n)
的额外空间,但这仍然不是最好的解决方案。- 你能想出一个仅使用常量空间的解决方案吗?
2.解题思路
2.1空间复杂度O(m+n)
使用两个数组,rows用于记录该行是否包含0,cols用于记录该列是否包含0,一次遍历后,rows,cols中分别保存了所有含0的行和列的情况,然后再遍历矩阵,给rows[i] = true 或 cols[j] = true的位置赋0即可,这种方式使用了O(m+n)的空间复杂度,可以进一步优化
2.2空间复杂度O(1)
上述思路是借助两个数组统计的某一行某一列的含0情况,而为了不使用额外的空间,我们可以就在matrix数组的第0列,第0行的位置记录 从[1,m-1)行中为0的情况,和[1,n-1)列中为0的情况,就相当于把rows,cols两个数组转化成了matrix的第0列和第0行,但是这么做会使得原来第0行第0列的情况没办法被记录下来,因此就需要两个flag标记位,先对第0行第0列是否含0进行记录,然后其他的位置 如 matrix[i][j]如果为0,就将matrix[0][j] 和 matrix[i][0]置为0,后面遍历遇到这个位置为0时,要把相应行和列的位置上也都置为0
3.代码实现
3.1方法一
/**
* 基于两个辅助数组rows cols,空间复杂度为O(m+n)
* @param matrix
*/
public void setZeroes(int[][] matrix) {
//统计哪一行 哪一列需要置为0
int m = matrix.length;
int n = matrix[0].length;
boolean[] rows = new boolean[m];
boolean[] cols = new boolean[n];
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (matrix[i][j] == 0) {
rows[i] = true;
cols[j] = true;
}
}
}
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (rows[i] || cols[j]) {
matrix[i][j] = 0;
}
}
}
}
3.2方法二
class Solution {
public void setZeroes(int[][] matrix) {
int m = matrix.length;
int n = matrix[0].length;
//flag1,flag2分别用于统计第0行,第0列是否含有0
boolean flag1 = false;
boolean flag2 = false;
for (int j = 0; j < n; j++) {
if (matrix[0][j] == 0) {
flag1 = true;
}
}
for (int i = 0; i < m; i++) {
if (matrix[i][0] == 0) {
flag2 = true;
}
}
//将第0行 和 第0列当作方法一中的rows 和 cols 辅助数组
for (int i = 1; i < m; i++) {
for (int j = 1; j < n; j++) {
if (matrix[i][j] == 0) {
matrix[i][0] = 0;
matrix[0][j] = 0;
}
}
}
//根据辅助数组中的值,给需要修改的地方赋值为0
for (int i = 1; i < m; i++) {
for (int j = 1; j < n; j++) {
if (matrix[i][0] == 0 || matrix[0][j] == 0) {
matrix[i][j] = 0;
}
}
}
//根据flag1,flag2的值,分别给第0行第0列赋0
if (flag1) {
for (int j = 0; j < n; j++) {
matrix[0][j] = 0;
}
}
if (flag2) {
for (int i = 0; i < m; i++) {
matrix[i][0] = 0;
}
}
}
}