leetCode 73.Set Matrix Zeroes (矩阵置0) 解题思路和方法

Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.

click to show follow up.

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;
                }
            }
        }
    }
}

作为一个完整的测试用例，这样的bug是不会留下的，所以上述代码光荣的牺牲了。

另外的方法就是需要一点技巧。将第一行和第一列作为标记，如果数组中有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;
            }
        }
    }
}