73. Set Matrix Zeroes

来源:互联网 发布:三星手表软件下载 编辑:程序博客网 时间:2024/06/05 16:52

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的话表示不换行。

原创粉丝点击