LeetCode 73. 矩阵置零

题目链接

73. 矩阵置零

用第一行与第一列来记录对应行是否有0
用两个变量来记录第一行和第一列是否有0

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        std::ios::sync_with_stdio(false);
        if(matrix.empty()) return ;
        int n = matrix.size(), m = matrix[0].size();
        int col = 1, row = 1;
        for(int i = 0; i < n; i++)
            if(!matrix[i][0]) col = 0;
        for(int i = 0; i < m; i++)
            if(!matrix[0][i]) row = 0;
        for(int i = 1; i < n; i++)
            for(int j = 1; j < m; j++)
                if(!matrix[i][j]){
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
        for(int i = 1; i < n; i++)
            if(!matrix[i][0])
                for(int j = 1; j < m; j++) matrix[i][j] = 0;
        for(int i = 1; i < m; i++)
            if(!matrix[0][i])
                for(int j = 1; j < n; j++) matrix[j][i] = 0;
        if(!col)
            for(int i = 0; i < n; i++) matrix[i][0] = 0;
        if(!row)
            for(int i = 0; i < m; i++) matrix[0][i] = 0;
    }
};