Leetcode 73. Set Matrix Zeroes Java版

解题思路:这道题很容易上手,是典型的矩阵操作问题,但是其中的陷阱是,我们遇到0的时候不能直接把矩阵的行列在当前矩阵直接置0,否则后面还没访问到的会被当成原来是0,最后会把很多不该置0的行列都置0了。我们看到其实判断某一项是不是0只要看它对应的行或者列应不应该置0就可以,所以我们可以维护一个行和列的判断数组,然后扫描一遍矩阵记录那一行或者列是不是应该置0即可,后面赋值是一个常量时间的判断。这种方法的空间复杂度是O(m+n)。

代码如下:

class Solution {
    public void setZeroes(int[][] matrix) {
        int m = matrix.length;
        int n = matrix[0].length;
        if(m==0 || n==0)
            return;
        int[] flagr = new int[m];
        int[] flagc = new int[n];
        
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                if(matrix[i][j]==0){
                    flagr[i] = 1;
                    flagc[j] = 1;
                }
            }
        }
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                if(flagr[i]==1 || flagc[j]==1)
                    matrix[i][j] = 0;
            }
        }
    }
}

猜你喜欢

转载自blog.csdn.net/qq_28900249/article/details/86604175