LeetCode-73. 矩阵置零-Java实现

题目链接

题目

结果

代码如下：

class Solution {
    public void setZeroes(int[][] matrix) {
        if(matrix==null||matrix.length<=0){
            return;
        }
        int rows = matrix.length;
        int columns = matrix[0].length;
        
        boolean zerox = false;
        boolean zeroy = false;
        int[] sigx = matrix[0];
        for(int i = 0;i < sigx.length;i++){
            if(sigx[i] == 0 ){
                zerox = true;
            }
        }
        for(int i = 0;i < rows;i++){
            if(matrix[i][0] == 0){
                zeroy = true;
            }
        }
        
    
        for(int i = 1;i < rows;i++){
            for(int j = 1 ; j < columns;j++ ){
                int target = matrix[i][j];
                if(target==0){
                    matrix[0][j] = 0;
                    matrix[i][0] = 0;
                }
            }
        }
        
        for(int i = 1;i < rows;i++){
            if(matrix[i][0]==0){
                for(int j = 1;j <columns;j++){
                    matrix[i][j] = 0;
                }
            }
        }
        
        for(int j = 1;j < columns;j++){
            if(matrix[0][j] == 0){
                for(int i = 1;i < rows;i++){
                    matrix[i][j] = 0;
                }
            }
        }
        
        if(zerox){
            for(int j = 0;j < columns;j++){
                matrix[0][j] = 0;
            }
        }
        if(zeroy){
            for(int i = 0;i < rows;i++){
                matrix[i][0] = 0;
            }
        }
    }
}

思路

代码有点丑陋，思路是用第一行和第一列做标记，比如（2，3）= 0 ,那么我们就把(2,0)=0,(0,3)=0，这样第一行和第一列就分别记录了各自维度为0的情况（这是第一行和第一列的0可能来自两种地方，一种是自己原来就是0，一种是对应的行或列中有值为0，但是效果是一样的，不论哪种情况，都是要将对应的列或行设置为0，所以不影响），但是要考虑，第一行或第一列如果本身存在0如何标记，毕竟（0，0）位置只有一个，所以我们使用另外的boolean来标记（也可以一个用boolean，一个用（0，0））。
总结一下步骤：先遍历第一行和第一列，进行0标记（标记1），然后对除去第一行和第一列的地方进行遍历，并在相应的位置（第一行和第一列）进行标记（标记2），最后依据标记1和标记2对矩阵的所有位置进行标记。