leetcode--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.

click to show follow up.
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?

题意：给定m*n的矩阵，如果某个元素是0，使其同行，同列的所有元素为0

要求使用O(m+n)空间复制度，甚至常数空间

解法1：首先遍历第一行，如果找到0,做标记，说明第一行要清0

遍历第一列，如果找到0,做标记，说明第一列要清0

然后从(1,1)开始遍历矩阵，找到0，将对应的第一行，第一列的元素设置为0，用于标记该行列应该清0

最后，根据第一行，第一列，0的情况，清0

public class Solution {
public void setZeroes(int[][] matrix) {
int rows = matrix.length;
int cols = matrix[0].length;
boolean firstRow=false,firstCol=false;
for(int i=0;i<rows;i++){
if(matrix[i][0]==0){
firstRow = true;
break;
}
}
for(int i=0;i<cols;i++){
if(matrix[0][i]==0){
firstCol = true;
break;
}
}
for(int i=1;i<rows;i++){
for(int j=1;j<cols;j++){
if(matrix[i][j]==0){
matrix[i][0] = 0;
matrix[0][j] = 0;
}
}
}
for(int i=1;i<rows;i++){
if(matrix[i][0]==0){
for(int j=1;j<cols;j++){
matrix[i][j] = 0;
}
}
}
for(int j=1;j<cols;j++){
if(matrix[0][j]==0){
for(int i=1;i<rows;i++){
matrix[i][j] = 0;
}
}
}
if(firstRow) for(int i=0;i<rows;i++) matrix[i][0] = 0;
if(firstCol) for(int j=0;j<cols;j++) matrix[0][j] = 0;
}
}