Offer 驾到，掘友接招！我正在参与2022春招打卡活动，点击查看活动详情。
一、题目描述：
题目来源：LeetCode-零矩阵
编写一种算法，若M × N矩阵中某个元素为0，则将其所在的行与列清零。
示例 1：
输入： [
[1,1,1],
[1,0,1],
[1,1,1]
]
输出： [
[1,0,1],
[0,0,0],
[1,0,1]
]
示例 2：
输入： [
[0,1,2,0],
[3,4,5,2],
[1,3,1,5]
]
输出： [
[0,0,0,0],
[0,4,5,0],
[0,3,1,0]
]
二、思路分析：
思路一：
- 可以使用两个数组，分别记录每一行和每一列是否有0出现
- 遍历数组，如果某个元素为0，则将元素对应的行和列在两个数组中标记
- 遍历完成后，再遍历新创建的两个数组，根据标记来决定是否需要更新原数组的值为0
思路二：
- 使用两个标记变量，分别记录第一行和第一列是否包含0
- 第一行和第一列判断完成后，整个矩阵也就判断完成了，最后只用根据标记变量进行矩阵的0值更新即可
三、AC 代码：
思路一：
class Solution {
public void setZeroes(int[][] matrix) {
int mLength = matrix.length;
int nLength = matrix[0].length;
boolean[] row = new boolean[mLength];
boolean[] col = new boolean[nLength];
for (int i = 0; i < mLength; i++) {
for (int j = 0; j < nLength; j++) {
if (matrix[i][j] == 0) {
row[i] = col[j] = true;
}
}
}
for (int i = 0; i < mLength; i++) {
for (int j = 0; j < nLength; j++) {
if (row[i] || col[j]) {
matrix[i][j] = 0;
}
}
}
}
}
思路二：
class Solution {
public void setZeroes(int[][] matrix) {
int m = matrix.length, n = matrix[0].length;
boolean flagCol0 = false, flagRow0 = false;
for (int i = 0; i < m; i++) {
if (matrix[i][0] == 0) {
flagCol0 = true;
}
}
for (int j = 0; j < n; j++) {
if (matrix[0][j] == 0) {
flagRow0 = true;
}
}
for (int i = 1; i < m; i++) {
for (int j = 1; j < n; j++) {
if (matrix[i][j] == 0) {
matrix[i][0] = matrix[0][j] = 0;
}
}
}
for (int i = 1; i < m; i++) {
for (int j = 1; j < n; j++) {
if (matrix[i][0] == 0 || matrix[0][j] == 0) {
matrix[i][j] = 0;
}
}
}
if (flagCol0) {
for (int i = 0; i < m; i++) {
matrix[i][0] = 0;
}
}
if (flagRow0) {
for (int j = 0; j < n; j++) {
matrix[0][j] = 0;
}
}
}
}
四、总结：
- 根据标记数组判断看似代码简单一点，实则运行逻辑更复杂一点
- 标记变量的话，运行效率更高，因为只用判断一行一列，最后只用更新即可
