【leetcode】Array—— Set Matrix Zeroes（73）

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(m+n)空间复杂度，开启了m+n长的标记数组，记录m行以及n列中是否存在0

代码：

    public void setZeroes(int[][] matrix) {

    int m = matrix.length;

    int n = matrix[0].length;

        int [] flag = new int [n+m];

        

        if(n==0||m==0)

        return;

        

        for(int i=0;i<m;i++){

        for(int j=0;j<n;j++){

        if(matrix[i][j]==0){

        flag[i]=1;

        flag[j+m]=1;

        }

        }

        }

        

        for(int i=0;i<m;i++){

        if(flag[i]==1){

        for(int j=0;j<n;j++)

        matrix[i][j]=0;

        }

        }

        

        for(int i=m;i<m+n;i++){

        if(flag[i]==1){

        for(int j=0;j<m;j++)

        matrix[j][i-m]=0;

        }

        }        

    }

改进：使用O(1)

遍历整个数组，发现0，把marix[0][j]以及marix[i][0]设为0，并使用两个标记变量，记录第一行和第一列是否有0

修改数组的时候，先修改内层(除了第一行和第一列)，然后根据两个标记变量修改第一行和第一列

代码：

public void setZeroes(int[][] matrix) {

    boolean fr = false,fc = false;

    for(int i = 0; i < matrix.length; i++) {

        for(int j = 0; j < matrix[0].length; j++) {

            if(matrix[i][j] == 0) {

                if(i == 0) fr = true;

                if(j == 0) fc = true;

                matrix[0][j] = 0;

                matrix[i][0] = 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(fr) {

        for(int j = 0; j < matrix[0].length; j++) {

            matrix[0][j] = 0;

        }

    }

    if(fc) {

        for(int i = 0; i < matrix.length; i++) {

            matrix[i][0] = 0;

        }

    }