Set Matrix Zeroes
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?
解题思路:
一开始没有看到constant space,后面进行了修改。
我个人的思路就是,在第一次遍历时,当遇到某个值为0的时候,将该行的第一个值和该列的第一个值置0,然后在第二次遍历时,如果某个数所在的行的第一个值为0或者所在的列的第一个值为0,就将该值置0。但是就有一个问题,就是,第一行和第一列是否需要全部置0的问题。所以在第二次遍历时,先不考虑第一行和第一列,在操作结束之后,再去考虑第一行和第一列。当然,再第一次遍历的时候先用boolean来存储是否需要将首行和首列置0的操作。
public class Solution {
public void setZeroes(int[][] matrix) {
if(matrix == null || matrix.length == 0 || matrix[0].length == 0)
return;
int rlen = matrix.length;
int clen = matrix[0].length;
boolean rr = false;
boolean cc = false;
for(int i = 0;i<rlen;i++)
{
for(int j = 0;j<clen;j++)
{
if(matrix[i][j] == 0)
{
matrix[i][0] = 0;
matrix[0][j] = 0;
if(i == 0) rr = true;
if(j == 0) cc = true;
}
}
}
for(int i = 1;i<rlen;i++)
{
for(int j = 1;j<clen;j++)
{
if(matrix[i][0] == 0 || matrix[0][j] == 0)
matrix[i][j] = 0;
}
}
if(cc == true)
{
for(int i = 1;i<rlen;i++)
matrix[i][0] = 0;
}
if(rr == true)
{
for(int j = 0;j<clen;j++)
matrix[0][j] = 0;
}
}
}