Set Matrix Zeroes

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?

Solution:

非常无聊的一道题。解题点就在于清空标志位存在哪里的问题。可以创建O(m+n)的数组来存储，但此题是希望复用已有资源。这里可以选择第一行和第一列来存储标志位。
所以使用constant space 其实是指复用空间而已。
1.先确定第一行和第一列是否需要清零
2.扫描剩下的矩阵元素，如果遇到了0，就将对应的第一行和第一列上的元素赋值为0
3.根据第一行和第一列的信息，已经可以讲剩下的矩阵元素赋值为结果所需的值了
4.根据1中确定的状态，处理第一行和第一列。

public class Solution {
    public void setZeroes(int[][] matrix) {
        // Start typing your Java solution below
        // DO NOT write main() function
        int row = 0;
        int col = 0;
        int sig = 0;
        if(matrix == null || matrix.length == 0 || matrix[0].length == 0) return;
        for(int i = 0; i < matrix.length; i ++){
            for(int j = 0; j < matrix[0].length; j ++){
                if(matrix[i][j] == 0){
                    if(sig == 0){
                        row = i;
                        col = j;
                        sig = 1;
                    }else{
                        matrix[row][j] = 0;
                        matrix[i][col] = 0;
                    }
                }
            }
        }
        if(sig == 1){
            for(int i = 0; i < matrix.length; i ++){
                for(int j = 0; j < matrix[0].length; j ++){
                if(matrix[i][col] == 0 || matrix[row][j] == 0){
                        matrix[i][j] = 0;
                    }
                }
            }
        }
    }
}

 网上的解法：

public class Solution {
    public void setZeroes(int[][] matrix) {
   
        boolean firstRow=false, firstColumn=false;
        for(int i=0;i<matrix.length;i++){
            if(matrix[i][0]==0){
                firstColumn = true;
                break;
            }
        }
        
        for(int i=0;i<matrix[0].length;i++){
            if(matrix[0][i]==0){
                firstRow = true;
                break;
            }
        }
    
        for(int i=1;i<matrix.length;i++){
            for(int j=1;j<matrix[0].length;j++){
                if(matrix[i][j]==0){
                    matrix[i][0]=0;
                    matrix[0][j]=0;
                }
            }
        }
        
        for(int i=1;i<matrix.length;i++){
            for(int j=1;j<matrix[0].length;j++){
                if(matrix[i][0]==0||matrix[0][j]==0){
                    matrix[i][j]=0;
                }
            }
        }
        
        if(firstRow){
            for(int i=0;i<matrix[0].length;i++)
                matrix[0][i]=0;
        }
        
        if(firstColumn){
            for(int i=0;i<matrix.length;i++)
                matrix[i][0]=0;
        }
        
    }
}

 第二遍：

找到哪些地方要设为0很容易，但是注意当变0的时候一定要注意顺序，不要后面又check之前改过的值，这样会把全部都变成0。

public class Solution {
    public void setZeroes(int[][] matrix) {
        // Start typing your Java solution below
        // DO NOT write main() function
        int row = 0;
        int col = 0;
        int sig = 0;
        if(matrix == null || matrix.length == 0 || matrix[0].length == 0) return;
        for(int i = 0; i < matrix.length; i ++)
            for(int j = 0; j < matrix[0].length; j ++)
                if(matrix[i][j] == 0){
                    if(sig == 0){
                        row = i;
                        col = j;
                        sig = 1;
                    }else{
                        matrix[row][j] = 0;
                        matrix[i][col] = 0;
                    }
                }
        if(sig == 1){
            for(int i = 0; i < matrix.length; i ++)
                if(matrix[i][col] == 0 && i != row)
                    for(int j = 0; j < matrix[0].length; j ++)
                        matrix[i][j] = 0;
            for(int i = 0; i < matrix[0].length; i ++)
                if(matrix[row][i] == 0)
                    for(int j = 0; j < matrix.length; j ++)
                        matrix[j][i] = 0;
            for(int j = 0; j < matrix[0].length; j ++)
                matrix[row][j] = 0;
        }
    }
}