描述
给定一个 m x n 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。
分析
- 用第0行和第0列标记除过第0行和第0列以外区域哪个点是0.
- 若第0行和第0列存在0,用两个boolean变量标记。
- 根据第0行和第0列上的0,把矩阵刷0.
- 根据两个boolean变量,把第0行和第0列刷0.
注意最好不要企图遍历一次matrix就完成1和2,这样容易漏掉结构特殊的矩阵。
class Solution {
public void setZeroes(int[][] matrix) {
boolean rowflag = false;
boolean colflag = false;
int row = matrix.length;
int col = matrix[0].length;
for(int i = 0; i < col; i++){
if(matrix[0][i] == 0){
rowflag = true;
}
}
for(int i = 0; i < row; i++){
if(matrix[i][0] == 0){
colflag = true;
}
}
for(int i = 1; i < row; i++){
for(int j = 1; j < col; j++){
if(matrix[i][j] == 0){
matrix[i][0] = 0;
matrix[0][j] = 0;
}
}
}
for(int i = 1; i < col; i++){
if(matrix[0][i] == 0){
for(int j = 0; j < row; j++){
matrix[j][i] = 0;
}
}
}
for(int i = 1; i < row; i++){
if(matrix[i][0] == 0){
for(int j = 0; j < col; j++){
matrix[i][j] = 0;
}
}
}
if(rowflag){
for(int i = 0; i < col; i++){
matrix[0][i] = 0;
}
}
if(colflag){
for(int i = 0; i < row; i++){
matrix[i][0] = 0;
}
}
}
}
虽然是一次遍历,但是要考虑的情况太多,容易出错。
class Solution {
public void setZeroes(int[][] matrix) {
boolean rowflag = false;
boolean colflag = false;
int row = matrix.length;
int col = matrix[0].length;
for(int i = 0; i < row; i++){
for(int j = 0; j < col; j++){
if(i == 0 && j == 0 && matrix[i][j] == 0){
rowflag = true;
colflag = true;
continue;
}
if(i == 0 && matrix[i][j] == 0){
rowflag = true;
continue;
}
if(j == 0 && matrix[i][j] == 0){
colflag = true;
continue;
}
if(matrix[i][j] == 0){
matrix[i][0] = 0;
matrix[0][j] = 0;
}
}
}
for(int i = 1; i < col; i++){
if(matrix[0][i] == 0){
for(int j = 0; j < row; j++){
matrix[j][i] = 0;
}
}
}
for(int i = 1; i < row; i++){
if(matrix[i][0] == 0){
for(int j = 0; j < col; j++){
matrix[i][j] = 0;
}
}
}
if(rowflag){
for(int i = 0; i < col; i++){
matrix[0][i] = 0;
}
}
if(colflag){
for(int i = 0; i < row; i++){
matrix[i][0] = 0;
}
}
}
}