Leetcode 73. Set Matrix Zeroes Java版

最新推荐文章于 2024-01-06 15:42:59 发布

原创最新推荐文章于 2024-01-06 15:42:59 发布 · 182 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode题解 Java版专栏收录该内容

11 篇文章

订阅专栏

本文介绍了一种解决矩阵置零问题的高效算法，避免了直接置零导致的数据错误。通过使用行和列的标记数组，可以准确地找出需要置零的元素，确保了算法的正确性和效率。

解题思路：这道题很容易上手，是典型的矩阵操作问题，但是其中的陷阱是，我们遇到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;
            }
        }
    }
}