一、问题描述
给定一个 m x n 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。
二、原地算法
什么是原地算法,就是空间复杂度很小,是o(1)或者是一个固定的常数。
三、解决思路
1、可以使用矩阵的第一行和第一列来标记这一行或者这一列是否存在0元素,但是要注意的是,我们要保持原来的0行0列是否有0元素,这个时候开辟两个额外的变量来记录一下。如果从0行0列开始标记的话,会出现的一种情况就是可能这个数组的这一行没有0元素,但是由于(0,0)是0,那么,(0,1)也会修改为0;(1,1)也因为(0,1)的修改而错误的改成0。如果用第一行第一列开始标记,不仅能判断第一行第一列是否为0,还能判断本行本列中是否有0。空间复杂度是o(1)。我的描述不是很清楚,自己跟着代码走一遍,就明白了。
2、可以使用两个set来记录,一个记录行标,一个记录列标,空间复杂度是 o(m+n);
三、代码
1、
public void setZeroes(int[][] matrix) {
//原地算法,只用o(1)的额外空间,或者很小的固定的空间
int row = matrix.length;
int col = matrix[0].length;
int flag_r = 0;
int flag_c = 0;
for (int i = 0; i < row; i++) {
if (matrix[i][0] == 0)
flag_r = 1;
}
for (int i = 0; i < col; i++) {
if (matrix[0][i] == 0)
flag_c = 1;
}
for (int i = 1; i < row; i++) {
for (int j = 1; j < col; j++) {
if (matrix[i][j] == 0) {
matrix[i][0] = matrix[0][j] = 0;
}
}
}
for (int i = 1; i < row; i++) {
for (int j = 1; j < col; j++) {
if (matrix[i][0] == 0 || matrix[0][j] == 0)
matrix[i][j] = 0;
}
}
if (flag_r == 1) {
for (int i = 0; i < row; i++) {
matrix[i][0] = 0;
}
}
if (flag_c == 1) {
for (int i = 0; i < col; i++) {
matrix[0][i] = 0;
}
}
}
2、
class Solution {
public void setZeroes(int[][] matrix) {
Set<Integer> row_zero = new HashSet<>();
Set<Integer> col_zero = new HashSet<>();
int row = matrix.length;
int col = matrix[0].length;
for (int i = 0; i < row; i++) {
for (int j = 0; j < col; j++) {
if (matrix[i][j] == 0) {
row_zero.add(i);
col_zero.add(j);
}
}
}
for (int i = 0; i < row; i++) {
for (int j = 0; j < col; j++) {
if (row_zero.contains(i) || col_zero.contains(j)) matrix[i][j] = 0;
}
}
}
}