[面试精选] 0073. 矩阵置零

1. 题目链接

73. 矩阵置零 - 力扣（LeetCode）

2. 题目描述

给定一个 m x n 的矩阵，如果一个元素为 0 ，则将其所在行和列的所有元素都设为 0 。使用 原地 算法

3. 题目示例

示例 1 :

输入：matrix = [[1,1,1],[1,0,1],[1,1,1]]
输出：[[1,0,1],[0,0,0],[1,0,1]]

示例 2 :

输入：matrix = [[0,1,2,0],[3,4,5,2],[1,3,1,5]]
输出：[[0,0,0,0],[0,4,5,0],[0,3,1,0]]

4. 解题思路

1. 问题理解：
   • 给定一个 m x n 的矩阵，如果某个元素为0，则将其所在的行和列的所有元素都设为0。
   • 要求使用原地算法，即不额外使用空间。
2. 关键思路：
   • 标记位法：
     • 使用第一行和第一列作为标记位，记录哪些行和列需要被置0。
     • 首先检查第一行和第一列是否包含0，并用两个布尔变量 row0_flag 和 col0_flag 记录。
     • 遍历除第一行和第一列外的所有元素，如果某个元素为0，则标记对应的第一行和第一列。
     • 根据标记位将对应的行和列置0。
     • 最后根据 row0_flag 和 col0_flag 处理第一行和第一列。
3. 算法流程：
   • 检查第一行和第一列是否包含0。
   • 使用第一行和第一列作为标记位，标记需要置0的行和列。
   • 根据标记位将对应的行和列置0。
   • 处理第一行和第一列。

5. 题解代码

class Solution {
    public void setZeroes(int[][] matrix) {
        int row = matrix.length;
        int col = matrix[0].length;
        
        // 标记第一行和第一列是否包含0
        boolean row0_flag = false;
        boolean col0_flag = false;
        
        // 检查第一行是否有0
        for (int j = 0; j < col; j++) {
            if (matrix[0][j] == 0) {
                row0_flag = true;
                break;
            }
        }
        
        // 检查第一列是否有0
        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) {
                    // 如果当前元素为0，则标记对应的第一行和第一列
                    matrix[i][0] = 0; // 标记第i行需要置0
                    matrix[0][j] = 0; // 标记第j列需要置0
                }
            }
        }
        
        // 根据标记位将对应的行和列置0
        for (int i = 1; i < row; i++) {
            for (int j = 1; j < col; j++) {
                // 如果第i行或第j列被标记为0，则将当前元素置0
                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;
            }
        }
    }
}

6. 复杂度分析

1. 时间复杂度：
   • 检查第一行和第一列：O(m) + O(n)。
   • 遍历矩阵标记需要置0的行和列：O(m * n)。
   • 根据标记位置0：O(m * n)。
   • 处理第一行和第一列：O(m) + O(n)。
   • 总体时间复杂度为 O(m * n)。
2. 空间复杂度：
   • 仅使用两个布尔变量 row0_flag 和 col0_flag，因此是 O(1)。