解题思路:
题目要求使用原地算法
在原地算法中,输入数据通常在内存中直接被修改,而不需要额外的辅助数据结构来存储中间结果。原地算法的一个常见应用是在数组或列表上进行操作,而不是创建新的数组或列表来存储结果。
java
class Solution {
public void setZeroes(int[][] matrix) {
int row = matrix.length; // 矩阵的行数
int col = matrix[0].length; // 矩阵的列数
boolean row0_flag = false; // 标记第一行是否有零
boolean col0_flag = false; // 标记第一列是否有零
// 检查第一行是否有零,遍历第一行需要知道有多少列
for (int j = 0; j < col; j++) {
if (matrix[0][j] == 0) {
row0_flag = true;
break;
}
}
// 检查第一列是否有零,遍历第一列需要知道有多少行
for (int i = 0; i < row; i++) {
if (matrix[i][0] == 0) {
col0_flag = true;
break;
}
}
// 使用第一行和第一列作为标志位
for (int i = 1; i < row; i++) {
for (int j = 1; j < col; j++) {
if (matrix[i][j] == 0) {
matrix[i][0] = matrix[0][j] = 0; // 如果元素为零,则将对应的第一行和第一列的元素设置为零
}
}
}
// 根据第一行和第一列的标志位,将矩阵中的元素置零
for (int i = 1; i < row; i++) {
for (int j = 1; j < col; j++) {
if (matrix[i][0] == 0 || matrix[0][j] == 0) {
matrix[i][j] = 0; // 如果第一行或第一列的元素为零,则将当前元素置零
}
}
}
// 如果第一行有零,则将第一行所有元素置零,遍历第一行需要知道有多少列
if (row0_flag) {
for (int j = 0; j < col; j++) {
matrix[0][j] = 0;
}
}
// 如果第一列有零,则将第一列所有元素置零,遍历第一列需要知道有多少行
if (col0_flag) {
for (int i = 0; i < row; i++) {
matrix[i][0] = 0;
}
}
}
}