Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.
Follow up:
Did you use extra space?
A straight forward solution using O(mn) space is probably a bad idea.
A simple improvement uses O(m + n) space, but still not the best solution.
Could you devise a constant space solution?
思路:此题最简单的方法当然就是另开一个数组,用来保存当前数组的状态。不过题意上是不推荐这种方法的,最好是用o(1)的空间。
我首先想的思路是如果为0,则行和列的非0项则全部职位Integer.MIN_VALUE,最后再将Integer.MIN_VALUE转换为0.但是这有个前提,就是数组中原数字不包含Integer.MIN_VALUE才行,不然就会出错。
代码如下:
public class Solution {
public void setZeroes(int[][] f) {
//思想是遇到0,将行和列不等于0的值全部设置为min
//最后再将min值替换为0
int min = Integer.MIN_VALUE;
for(int i = 0; i < f.length; i++){
for(int j = 0; j < f[0].length; j++){
if(f[i][j] == 0){
//替换行
for(int x = 0; x < f[0].length;x++){
if(f[i][x] != 0)//不等于0才替换
f[i][x] = min;
}
//替换列
for(int x = 0; x < f.length;x++){
if(f[x][j] != 0)//不等于0才替换
f[x][j] = min;
}
break;//进行下一行
}
}
}
//替换回0
for(int i = 0; i < f.length; i++){
for(int j = 0; j< f[0].length;j++){
if(f[i][j] == min){
f[i][j] = 0;
}
}
}
}
}
另外的方法就是需要一点技巧。将第一行和第一列作为标记,如果数组中有f[i][j]为0,则将f[i][0]和f[0][j]标记为0.最后再根据标记转换。
代码如下:
public class Solution {
public void setZeroes(int[][] f) {
if(f.length == 0 || f[0].length == 0)
return;
boolean isR0 = false;
boolean isC0 = false;
//第一列
for(int i = 0; i < f.length; i++){
if(f[i][0] == 0){//第一列如果有0
isC0 = true;
}
}
//第一行
for(int i = 0; i < f[0].length; i++){
if(f[0][i] == 0){//第一行如果有0
isR0 = true;
}
}
//遍历
for(int i = 0; i < f.length; i++){
for(int j = 0; j < f[0].length;j++){
if(f[i][j] == 0){
f[i][0] = 0;//第一列记录
f[0][j] = 0;//第一行记录
}
}
}
//根据第一列填充行
for(int i = 1; i < f.length; i++){
if(f[i][0] == 0){//第一列
for(int j = 0; j < f[0].length; j++){
f[i][j] = 0;
}
}
}
//根据第一行填充列
for(int i = 1; i < f[0].length; i++){
if(f[0][i] == 0){
for(int j = 0; j < f.length; j++){
f[j][i] = 0;
}
}
}
if(isR0){//第一行也为0
for(int i = 0; i < f[0].length; i++){
f[0][i] = 0;
}
}
if(isC0){//第一列也为0
for(int i = 0; i < f.length; i++){
f[i][0] = 0;
}
}
}
}