面试题 01.08. 零矩阵
编写一种算法,若M × N矩阵中某个元素为0,则将其所在的行与列清零。
示例 1:
输入:
[
[1,1,1],
[1,0,1],
[1,1,1]
]
输出:
[
[1,0,1],
[0,0,0],
[1,0,1]
]
题解
class Solution {
public void setZeroes(int[][] matrix) {
boolean isFirstRowHaveZero = false;
boolean isFirstColHaveZero = false;
// 判断第一行是否有0
for(int i = 0; i < matrix.length; i++) {
if (matrix[i][0] == 0) {
isFirstColHaveZero = true;
}
}
// 判断第一列是否有0
for(int j = 0; j < matrix[0].length; j++) {
if (matrix[0][j] == 0) {
isFirstRowHaveZero = true;
}
}
// 利用第一行和第一列记录0位置
for(int i = 1; i < matrix.length; i++) {
for(int j = 1; j < matrix[i].length; j++) {
if (matrix[i][j] == 0) {
matrix[0][j] = 0;
matrix[i][0] = 0;
}
}
}
// 对除第一列和第一行的元素清零
for(int i = 1; i < matrix.length; i++) {
for(int j = 1; j < matrix[i].length; j++) {
if (matrix[0][j] == 0 || matrix[i][0] == 0) {
matrix[i][j] = 0;
}
}
}
// 对除第一行元素清零
for(int i = 0; i < matrix.length; i++) {
if (isFirstColHaveZero) {
matrix[i][0] = 0;
}
}
// 对除第一列元素清零
for(int j = 0; j < matrix[0].length; j++) {
if (isFirstRowHaveZero) {
matrix[0][j] = 0;
}
}
}
}