力扣Hot100-73矩阵置零(标记数组)

给定一个 m x n 的矩阵,如果一个元素为 ,则将其所在行和列的所有元素都设为 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 == matrix.length
  • n == matrix[0].length
  • 1 <= m, n <= 200
  • -231 <= matrix[i][j] <= 231 - 1

进阶:

  • 一个直观的解决方案是使用  O(mn) 的额外空间,但这并不是一个好的解决方案。(查找到一个就遍历一次行和列)
  • 使用O(m+n)的方法,如下:

思路:使用两个标记数组(哈希表记录)row,col分别记录二维矩阵中值为零的元素所在的行列值。再遍历一遍矩阵,判断当前元素值是否位于被标记的行或列上,位于则直接值为零。

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        map<int,int>row,col;
        for(int i=0;i<matrix.size();i++){//row
        vector<int>a;
        a=matrix[i];
        for(int j=0;j<a.size();j++){//column
            if(a[j]==0){
                row[i]=1;
                col[j]=1;

            }
        }
        }
        for(int i=0;i<matrix.size();i++){
            vector<int>a;
             a=matrix[i];
            for(int j=0;j<a.size();j++){
                if(row[i]==1||col[j]==1){//位于有0的行或列的数,所以值为零
                matrix[i][j]=0;

                }
            }
        }

    }
};
  • 一个简单的改进方案是使用 O(m + n) 的额外空间,但这仍然不是最好的解决方案。(注意时间复杂度仍然是O(m*n)),空间复杂度由O(m+n)降低到O(1));
  • 你能想出一个仅使用常量空间的解决方案吗?

用矩阵的第一行和第一列代替方法一中的两个标记数组,以达到 O(1)的额外空间。但这样会导致原数组的第一行和第一列被修改,无法记录它们是否原本包含 0。因此我们需要额外使用两个标记变量分别记录第一行和第一列是否原本包含 0。

在实际代码中,我们首先预处理出两个标记变量,接着使用其他行与列去处理第一行与第一列,然后反过来使用第一行与第一列去更新其他行与列,最后使用两个标记变量更新第一行与第一列即可。

注意,在使用第一行和第一列记录其他位置是否存在值为零的元素时,若存在应该将第一行和第一列对应的值置为,(若第一行中存在零,则对应的这一列也要值为零,所以也会被标记,若第一行不存在零也不会导致影响第一行原本存在的值)。

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        //首先,使用两个变量分别记录第一行和第一列是否存在值为零的元素
        int a=0,b=0;
         vector<int>row;
        row=matrix[0];
        for(int i=0;i<row.size();i++){//第一列
            if(row[i]==0){
                a=1;
                break;
            }
        }
        for(int i=0;i<matrix.size();i++){//第一行
            if(matrix[i][0]==0){
                b=1;
                break;
            }
        }
        //使用第一行和第一列分别记录其他数组中
        for(int i=1;i<matrix.size();i++){//行
              vector<int>c;
            c=matrix[i];
            for(int j=1;j<c.size();j++){//列
                if(matrix[i][j]==0){
                    matrix[0][j]=0;//行
                    matrix[i][0]=0;//列

                }
            }
        }

        //遍历一遍,判断第一行第一列以外的元素是否需要置零
        for(int i=1;i<matrix.size();i++){
             vector<int>c;
            c=matrix[i];
            for(int j=1;j<c.size();j++){
                if(matrix[i][0]==0 || matrix[0][j]==0){
                    matrix[i][j]=0;//注意写成==导致错误
                }

            }

        }
        if(a==1){
            for(int i=0;i<row.size();i++){
                matrix[0][i]=0;
            }
        }
        if(b==1){
            for(int j=0;j<matrix.size();j++){
                matrix[j][0]=0;
            }
        }



    }
};

  • 4
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

半截詩

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值