leetCode练习(73)

来源:互联网 发布:数据库题库 编辑:程序博客网 时间:2024/04/30 07:30

题目:Submission Details

难度:medium

Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.

Follow up:

Did you use extra space?
A straight forward solution using O(mn) space is probably a bad idea.
A simple improvement uses O(m + n) space, but still not the best solution.
Could you devise a constant space solution?

解题思路:给一个矩阵,如果matrix[I][j]=0,则让该行该列的数全置0.

看到此题,第一想法是遍历矩阵,如果遇到0,则行列置0.这是错误的解法!比如:

一个矩阵0 0 0 5
              4 3 1 4
              0 1 1 4
              1 2 1 3 
              0 0 1 1

遇到第一个0,将b[1][0]变成了0,等遍历到b[1][0],b[1][0]的值已经变了,解题也会出错。

第二个想法是构建一个平行矩阵matrix2[I][j],遍历matrix[I][j],然后将变化填入matrix2[I][j]中即可。但是使用了额外空间,占用了额外O(n*m)内存。

接着我想到了之前看到的N皇后问题的优化解,可以使用两个数组,hang[row]和lie[column]来表示矩阵的第row行,第column是否被置0,这样就只用了O(m+n),简便了许多。

具体代码如下:

public static void setZeroes(int[][] matrix) {        int row=matrix.length;        int column=matrix[0].length;        boolean[] hang=new boolean[row];        boolean[] lie=new boolean[column];        for(int i=0;i<row;i++){        for(int j=0;j<column;j++){        if(matrix[i][j]==0){        hang[i]=true;        lie[j]=true;        }        }        }        for(int i=0;i<row;i++){        if(hang[i]){        for(int j=0;j<column;j++){        matrix[i][j]=0;        }        }        }        for(int i=0;i<column;i++){        if(lie[i]){        for(int j=0;j<row;j++){        matrix[j][i]=0;        }        }        }    }


0 0
原创粉丝点击