leetcode--set_matrix_zeroes

leetcode–set_matrix_zeroes

题意： 给定一个m×n的二维数组，如果有一个元素是0就将该元素所在的行和列全部设置成0。
条件： 要求就地转换，不使用额外空间。
分析： 首先这道题的难点就是不引入额外空间，如果可以引入$O(m+n)$ 的空间，那么就可以使用两个集合分别存横坐标和纵坐标，如果某个元素是0，就将该行所在的横坐标和纵坐标分别添加到集合当中。因为集合中不能有重复的元素，所以空间复杂度刚好是$O(m+n)$ 。
               就地转换借鉴了该思想，使用第0行和第0列的空间做记录。如果(i, j)处元素是0，则设置(i, 0)为0、(0, j)为0。最后遍历第0行和0列，如果现有元素值是0，就将对应的行或列全部设置为0，不过在所有计算之前需要先遍历第0行和0列，判断原来的元素中是否有0。（代码中有相关的注释）

代码

//利用第1行和第1列的空间做记录
    public void setZeroes(int[][] matrix) {
        //遍历第0行
        int row_length = matrix.length;         //行数
        int col_length = matrix[0].length;      //列数
        boolean row0 = false;                   //记录第0行是否有0
        boolean col0 = false;                   //记录第0列是否有0
        if(row_length <= 0 || col_length <= 0){ 
            return;
        }else if(row_length == 1 || col_length == 1){   //只有1行或者1列时
            for(int i = 0; i < row_length; i++){
                for(int j = 0; j < col_length; j++){
                    if(matrix[i][j] == 0){
                        row0 = true;
                        break;
                    }
                }
            }
            if(row0 == true){
                for(int i = 0; i < row_length; i++){
                    for(int j = 0; j < col_length; j++){
                        matrix[i][j] = 0;
                    }
                }
            }
        }else{                                          //行列都大于1时，判断第0行和第0列是否有0
            for(int j = 0; j < col_length; j++){
                if(matrix[0][j] == 0){
                    row0 = true;
                    break;
                }
            }
            for(int i = 0; i < row_length; i++){
                if(matrix[i][0] == 0){
                    col0 = true;
                    break;
                }
            }

            //遍历整个数组，判断是否为0
            for(int i = 1; i < row_length; i++){
                for(int j = 1; j < col_length; j++){
                    if(matrix[i][j] == 0){
                        matrix[i][0] = 0;               //将该结果保存到[i, 0]相应位置
                        matrix[0][j] = 0;               //将该结果保存到[0, j]相应位置
                    }
                }
            }

            //遍历第0行和第0列，将0所在的行或者列设置为0
            for(int i = 1; i < row_length; i++){
                if(matrix[i][0] == 0){
                    for(int j = 1; j < col_length; j++){
                        matrix[i][j] = 0;
                    }
                }
            }
            for(int j = 1; j < col_length; j++){
                if(matrix[0][j] == 0){
                    for(int i = 1; i < row_length; i++){
                        matrix[i][j] = 0;
                    }
                }
            }

            //最后改变第0行和第0列
            if(row0){
                for(int j = 0; j < col_length; j++){
                    matrix[0][j] = 0;
                }
            }
            if(col0){
                for(int i = 0; i < row_length; i++){
                    matrix[i][0] = 0;
                }
            }
        }
    }

源程序下载