leetcode热题HOT 73. 矩阵置零

一、问题描述：

标记需要设置为 0 的行和列：遍历整个矩阵，如果当前元素为 0，则将对应的行首和列首元素置为 0，用于记录该行和该列需要设置为 0。

二、解题思路：

通过记录需要置零的行和列索引来实现：

1. 使用两个列表 setc 和 setk 分别记录包含 0 的行和列的索引。
   遍历整个矩阵，如果当前元素为 0，则将该元素所在的行和列的索引分别添加到对应的列表中。
   将记录下来的行和列设置为 0：遍历记录的行和列的列表，将对应的行或列全部设置为 0。
2. 空间优化：
   ①记录每行和每列是否包含 0：用两个变量 flagCol0 和 flagRow0 记录第一列和第一行是否包含 0 元素。然后使用首行 matrix[i][0] 和首列 matrix[0][j] 两个列表代替 setc 和 setk 记录包含 0 的行和列的索引。
   ②标记需要设置为 0 的行和列：遍历整个矩阵，若当前元素为 0，则将对应的行首和列首元素置为 0，用于记录该行和该列需要设置为 0。
   ③根据标记设置行和列为 0：再次遍历整个矩阵，如果当前行首或列首元素为 0，则将对应元素设置为 0。判断第一行和第一列是否包含0，若包含，也需要置零。

三、代码示例：

1. 使用标记列表

class Solution {
    public void setZeroes(int[][] matrix) {
        int m = matrix.length; // 获取矩阵的行数
        int n = matrix[0].length; // 获取矩阵的列数
        List<Integer> setc = new ArrayList<>(); // 创建一个列表用于存放需要置零的行索引
        List<Integer> setk = new ArrayList<>(); // 创建一个列表用于存放需要置零的列索引
        // 遍历整个矩阵，找出值为 0 的元素所在的行列索引
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                if(matrix[i][j] == 0){
                    setc.add(i); // 将需要置零的行索引添加到 setc 列表中
                    setk.add(j); // 将需要置零的列索引添加到 setk 列表中
                }
            }
        }
        // 将 setc 中记录的行索引对应的整行元素全部置零
        for (Integer i : setc){
            for(int j = 0; j < n; j++) 
                matrix[i][j] = 0;
        }
        // 将 setk 中记录的列索引对应的整列元素全部置零
        for (Integer j : setk){
            for(int i = 0; i < m; i++) 
                matrix[i][j] = 0;
        }
    }
}

• 时间复杂度分析：
  遍历整个矩阵来记录包含 0 的行和列的时间复杂度为 O(mn)。
  第二次遍历矩阵来设置为 0 的行和列的时间复杂度也为 O(mn)。
• 空间复杂度分析：使用了两个列表来记录行和列的索引，最坏情况下需要 O(m + n) 的额外空间来存储这些索引，因此空间复杂度为 O(m + n)。

2.空间优化，使用两个标记变量

class Solution {
    public void setZeroes(int[][] matrix) {
        int m = matrix.length; // 获取矩阵的行数
        int n = matrix[0].length; // 获取矩阵的列数
        boolean flagCol0 = false; // 标记第一列是否需要置零
        boolean flagRow0 = false; // 标记第一行是否需要置零
        
        // 检查第一列是否存在 0 元素，若存在则将 flagCol0 置为 true
        for (int i = 0; i < m; i++) {
            if (matrix[i][0] == 0) 
                flagCol0 = true;
        }
        
        // 检查第一行是否存在 0 元素，若存在则将 flagRow0 置为 true
        for (int j = 0; j < n; j++) {
            if (matrix[0][j] == 0) 
                flagRow0 = true;
        }
        // 遍历整个矩阵，如果某个元素为 0，则将该元素所在行和列的首元素置为 0，作为标记
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                if(matrix[i][j] == 0){
                    matrix[0][j] = 0; // 将该元素所在列的首元素置为 0
                    matrix[i][0] = 0; // 将该元素所在行的首元素置为 0
                }
            }
        }
        // 根据标记的首元素，将对应行列的元素置为 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; // 将对应行列的元素置为 0
                }
            }
        }
        // 根据标记的标志，将第一列和第一行的元素置为 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; 
        }
    }
}

• 时间复杂度分析：
  遍历矩阵来标记需要设置为 0 的行和列的时间复杂度为 O(mn)。
  第二次遍历矩阵来设置为 0 的行和列的时间复杂度也为 O(mn)。
  总体时间复杂度为 O(mn)。
• 空间复杂度分析：
  除了使用几个常量大小的额外空间外，算法并未使用额外的空间，因此空间复杂度为 O(1)。