signed

QiShunwang

“诚信为本、客户至上”

leetcode_73_矩阵置零

2021/3/21 10:16:06   来源:

leetcode_73_矩阵置零

给定一个 m x n 的矩阵,如果一个元素为 0,则将其所在行和列的所有元素都设为 0。请使用原地算法。

class Solution {
    public void setZeroes(int[][] matrix) {
        //新思路,使用矩阵的第一行和第一列来存储矩阵该行和该列的某些重要特性
        //在本题中,使用第一行和第一列来记载该列是否有元素为零,而第一列是否有零使用一个布尔变量来记载
        int m = matrix.length;
        int n = matrix[0].length;
        boolean flagcol1 = false;
        for(int i=0;i<m;i++){
            if(matrix[i][0]==0){
                    flagcol1 = true;
                }
            for(int j=1;j<n;j++){
                
                if(matrix[i][j]==0){
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        }

        //因为第一行是标记位,我们为了不提前更改标记位的值,需要从最后一行开始遍历,而第一列因为使用了额外的
        //布尔变量来存储该列是否有零,因此不用遍历,只需要检测完该行后根据需要将第一列的相应行变零即可

        for(int i=m-1;i>=0;i--){
            for(int j=1;j<n;j++){
                if(matrix[i][0]==0||matrix[0][j]==0){
                    matrix[i][j]=0;
                }
                
            }
            if(flagcol1){
                    matrix[i][0]=0;
            }
        }





    }
}