LeetCode 73. 矩阵置零

原题链接

题目要求

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

进阶：

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

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

思路

• $O(n+m)$复杂度的思路为开两个辅助数组，长度分别为n和m，用来标记哪些行和哪些列需要置为0，而题目要求的原地算法是不借助辅助空间，也即是在原数组空间的基础上进行操作

• 将以下红色的部分作为辅助数组，标记对应行/列是否应该置零
  

• 遍历蓝色部分的矩阵，如果找到一个数字是0，那么将对应行与列红色区域元素置为0

• 然后，遍历红色部分，检查是否为0，从而确定对应行/列是否应该置零

• 最后，特判r和c，从而得出第0行和第0列是否需要置零（r:标记第0行 c：标记第0列）

代码

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int r=1,c=1;//r:标记第0行  c：标记第0列
        
        int n=matrix.size();//行
        int m=matrix[0].size();//列
        
        for(int i=0;i<m;i++)
            if(matrix[0][i]==0) r=0;//如果第一行中有为0的数字，则r置为0
        for(int i=0;i<n;i++) 
            if(matrix[i][0]==0) c=0;//如果第一列中有为0的数字，则c置为0
        
        for(int i=1;i<n;i++)
            for(int j=1;j<m;j++)//两个for循环遍历除了第一行和第一列之外整个矩阵
                if(matrix[i][j]==0)//如果某个数字是0，那么将对应第一行和第一列元素标记为0
                {
                    matrix[i][0]=0;
                    matrix[0][j]=0;
                }
        for(int i=1;i<m;i++)
            if(matrix[0][i]==0) //如果第0行第i列的数字为0，那么
                for(int j=1;j<n;j++)
                    matrix[j][i]=0;
        for(int i=1;i<n;i++)
            if(matrix[i][0]==0) 
                for(int j=1;j<m;j++)
                    matrix[i][j]=0;
        
        if(!r)//如果r==0，则第一行全置为0
            for(int i=0;i<m;i++)
                matrix[0][i]=0;
        if(!c)//如果c==0，则第一列全置为0
            for(int i=0;i<n;i++)
                matrix[i][0]=0;
        
     }
};