LeetCode73. 矩阵置零（2024秋季每日一题 19）

sweetheart7-7

已于 2024-09-18 21:50:14 修改

阅读量737

点赞数 24

分类专栏：秋季每日一题2024 文章标签：矩阵线性代数数组 leetcode

于 2024-09-18 21:22:34 首次发布

本文链接：https://blog.csdn.net/qq_46456049/article/details/142329733

版权

秋季每日一题2024 专栏收录该内容

42 篇文章 0 订阅

订阅专栏

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

示例 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]]

提示：

$m == ma t r i x . l e n g t h$
$n == ma t r i x [0] . l e n g t h$
$1 <= m, n <= 200$
$2^{31} <= matrix[i][j] <= 2^{31} - 1$

进阶：

一个直观的解决方案是使用 O(mn) 的额外空间，但这并不是一个好的解决方案。
一个简单的改进方案是使用 O(m + n) 的额外空间，但这仍然不是最好的解决方案。
你能想出一个仅使用常量空间的解决方案吗？

思路：

首先，可以考虑用两个数组，分别记录当前列和当前行是否有 0，空间复杂度为 $O (m + n)$
其次，在上面的基础上，可以考虑用原数组的第一行和第一行充当上面的数组，去记录
然而，这样会丧失第一列/第一行是否有 0 的信息
可以用 flag_row，flag_colu 两个变量去记录第一列/行是否有0
此时的空间复杂度为 $O (1)$

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        bool flag_row = false, flag_colu = false;
        int n = matrix.size(), m = matrix[0].size();
        for(int i = 0; i < n; i++)
            for(int j = 0; j < m; j++){
                if(matrix[i][j] == 0){
                    matrix[0][j] = 0, matrix[i][0] = 0;
                    if(i == 0) flag_row = true;
                    if(j == 0) flag_colu = true;
                }
            }
        for(int i = n - 1; i >= 0; i--)
            for(int j = m - 1; j >= 0; j--){
                if(matrix[i][0] == 0 || matrix[0][j] == 0){
                    if(i && j) matrix[i][j] = 0;
                    if(i == 0 && flag_row)  matrix[i][j] = 0;
                    if(j == 0 && flag_colu) matrix[i][j] = 0;
                }
            }
    }
};