73. Set Matrix Zeroes

73. Set Matrix Zeroes

题目

给出一个m x n 的矩阵，如果有元素是零，将它所在的行和列都设为0。在原数组上进行变换。
继续看：
你使用额外的空间了吗？
一个比较直接的方法是使用O(mn)空间，复杂度较高。
一个简单的改进是使用O(m + n)空间，但是仍然不是最好的解决办法。
请设计一个常数空间的方法。

代码块

class Solution {
   public static void setZeroes(int[][] matrix){
        boolean fr = false ,  fc = false; //row column flag
        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;
            }

        }

    }

      public static void main(String[] args) {
            int[][] matrix = {{1,2,3},
                              {4,0,6},
                              {7,8,9}};
            setZeroes(matrix);
            for(int i = 0; i < 3;i++){
                for(int j = 0; j < 3; j++){
                    System.out.print(matrix[i][j] + " ");
                }
                System.out.println(); //打印换行
            }

        }
}

代码解析

本题要求的是空间复杂度，而不是时间复杂度。
1. 首先在行列设置flag，如果发现零元素，并且该零元素位于第一行或者第一列，然后触发事件（flag= true 跳到3）；并将第一行或者第一列的相应元素记为0；
2. 然后将再除去第一行第一列，开始遍历数组，如果对应的第一行和第一列有零，则将其置为0；
3. 最后根据第一行第一列的flag来更新第一行第一列。
注：
更新flag时使用的判断条件是索引而非矩阵元素；
matrix.length表示的行的长度；
matrix[0].length表示的是列的长度；
JAVA中字符的加号表示连接号；
System.out.println(); 表示打印换行，不加ln的话表示不换行。