Leetcode 73. 矩阵置零

最新推荐文章于 2024-07-24 11:56:55 发布

自信的小螺丝钉

最新推荐文章于 2024-07-24 11:56:55 发布

阅读量136

点赞数

分类专栏： Leetcode 文章标签： leetcode 矩阵算法数据结构 c++

本文链接：https://blog.csdn.net/qq_45791939/article/details/128057168

版权

Leetcode 专栏收录该内容

325 篇文章 0 订阅

订阅专栏

原题链接：Leetcode 73. 矩阵置零

在这里插入图片描述

数组标记：O(m+n)

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int m=matrix.size();
        int n=matrix[0].size();
        vector<int> row(m);
        vector<int> col(n);
        for(int i=0;i<m;i++)
        {
            for(int j=0;j<n;j++)
            {
                if(matrix[i][j]==0)
                {
                    row[i]=1; col[j]=1;
                }
            }
        }
        for(int i=0;i<m;i++)
        {
            for(int j=0;j<n;j++)
            {
                if(row[i] || col[j] ) matrix[i][j]=0;
            }
        }
    }
};

变量标记：O(1)

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int m=matrix.size();
        int n=matrix[0].size();
        int col0=0,row0=0;
        for(int i=0;i<m;i++)
        {
            if(!matrix[i][0]) col0=1;
        }
        for(int j=0;j<n;j++)
        {
            if(!matrix[0][j]) row0=1;
        }
        //排除第一行和第一列，防止错误更新
        for(int i=0;i<m;i++)
        {
            for(int j=0;j<n;j++)
            {
                if(matrix[i][j]==0)
                {
                    matrix[i][0]=0;
                    matrix[0][j]=0;
                }
            }
        }
        for(int i=1;i<m;i++)
        {
            for(int j=1;j<n;j++)
            {
                if(!matrix[i][0] || !matrix[0][j]) matrix[i][j]=0;
            }
        }
        if(col0)
        {
            for(int i=0;i<m;i++) matrix[i][0]=0;
        }
        if(row0)
        {
            for(int j=0;j<n;j++) matrix[0][j]=0;
        }
    }
};