162. 矩阵归零

162. 矩阵归零

 

给定一个m×n矩阵，如果一个元素是0，则将其所在行和列全部元素变成0。

需要在原矩阵上完成操作。

样例

样例 1:

输入:[[1,2],[0,3]]

输出:[[0,2],[0,0]]

样例 2:

输入:[[1,2,3],[4,0,6],[7,8,9]]

输出:[[1,0,3],[0,0,0],[7,0,9]]

挑战

 

你是否使用了额外的空间？

一个直接的解决方案是使用 O(MN) 的额外空间，但这并不是一个好的方案。

一个简单的改进方案是使用 O(M + N) 的额外空间，但这仍然不是最好的解决方案。

你能想出一个常数空间的解决方案吗？

 

 

public class Solution {

    /**

     * @param matrix: A lsit of lists of integers

     * @return: nothing

     */

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

      if (matrix.length == 0) return;

        boolean iflag = false;

        boolean jflag = false;

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

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

                iflag = true;

                break;

            }

        }

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

            if (matrix[0][m] == 0) {

                jflag = true;

                break;

            }

        }

 

 

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

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

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

                    matrix[0][j] = 0;

                    matrix[i][0] = 0;

                }

            }

        }

 

 

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

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

                Arrays.fill(matrix[i], 0);

            }

        }

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

            if (matrix[0][m] == 0) {

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

                    matrix[i][m] = 0;

                }

            }

        }

        if (iflag) {

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

                matrix[i][0] = 0;

            }

        }

        if (jflag) {

            Arrays.fill(matrix[0], 0);

        }

    }

}