解题思路:
题目要求使用原地算法
在原地算法中,输入数据通常在内存中直接被修改,而不需要额外的辅助数据结构来存储中间结果。原地算法的一个常见应用是在数组或列表上进行操作,而不是创建新的数组或列表来存储结果。
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;}} }
}