73_矩阵置零

73.矩阵置零

题目描述:给定一个 *m* x *n* 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0


自己的方案:

class Solution {
    public void setZeroes(int[][] matrix) {
        int m=matrix.length;
        int n=matrix[0].length;
        List<Integer> list=new ArrayList<>(10);
        for(int i=0;i<m*n;i++){
            if(matrix[i/n][i%n]==0){
                list.add(i);
            }
        }
        for(int num:list){
            for(int i=0;i<m;i++){
                matrix[i][num%n]=0;
            }
            for(int i=0;i<n;i++){
                matrix[num/n][i]=0;
            }
        }
    }
}

遍历数组,用list记录值为零的位置,这里记录的是0到mn之间的整数,原因是使用了一种表示二维数组元素位置的替换法:任意x(0到mn)对应二维数组中第x/n行,第x%n列;这种方法在566题重塑矩阵那里详细介绍了。

然后取出记录,把每个为0的位置所在行和所在列置为0。


官方优化方案:

class Solution {
    public void setZeroes(int[][] matrix) {
        int m = matrix.length, n = matrix[0].length;
        boolean flagCol0 = false, flagRow0 = false;
        for (int i = 0; i < m; i++) {
            if (matrix[i][0] == 0) {
                flagCol0 = true;
            }
        }
        for (int j = 0; j < n; j++) {
            if (matrix[0][j] == 0) {
                flagRow0 = true;
            }
        }
        for (int i = 1; i < m; i++) {
            for (int j = 1; j < n; j++) {
                if (matrix[i][j] == 0) {
                    matrix[i][0] = matrix[0][j] = 0;
                }
            }
        }
        for (int i = 1; i < m; i++) {
            for (int j = 1; j < n; j++) {
                if (matrix[i][0] == 0 || matrix[0][j] == 0) {
                    matrix[i][j] = 0;
                }
            }
        }
        if (flagCol0) {
            for (int i = 0; i < m; i++) {
                matrix[i][0] = 0;
            }
        }
        if (flagRow0) {
            for (int j = 0; j < n; j++) {
                matrix[0][j] = 0;
            }
        }
    }
}

作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/set-matrix-zeroes/solution/ju-zhen-zhi-ling-by-leetcode-solution-9ll7/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

原方法用列表记录位置,需要开辟额外的空间,优化方法用第一行和第一列记录为0的位置,无额外空间开销,但这样会改变第一行和第一列的数据,所以要用两个变量来标记第一行是否含有0,第一列是否含有0。

上一篇:924891-73-6 苯甲酸,4,4‘-(21H,23H-卟吩-5,15-二酰基)双-1,1‘-二乙酯


下一篇:SublimeText安装中文、英文字体