Leetcode 73. 矩阵置零

原题链接：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;
        }
    }
};