LeetCode 73. 矩阵置零_矩阵置零 c语言leetcode

思路：用数组的第一行记录这一列有没有零，第一列记录这一行有没有零，r0,c0记录第一行第一列有没有0

代码：

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        if(matrix.empty()) return ;
        int n=matrix.size(),m=matrix[0].size();
        int r0=1,c0=1;
        for(int i=0;i<m;i++) if(!matrix[0][i]) r0=0;
        for(int i=0;i<n;i++) if(!matrix[i][0]) c0=0;

        for(int i=1;i<m;i++)
            for(int j=0;j<n;j++)
                if(!matrix[j][i])
                    matrix[0][i]=0;
        
        for(int i=1;i<n;i++)
            for(int j=0;j<m;j++)
                if(!matrix[i][j])
                    matrix[i][0]=0;

        for(int i=1;i<m;i++)
            if(!matrix[0][i])
            for(int j=0;j<n;j++)
                matrix[j][i]=0;

        for(int i=1;i<n;i++)
            if(!matrix[i][0])
                for(int j=0;j<m;j++)
                    matrix[i][j]=0;

        if(!r0) for(int i=0;i<m;i++) matrix[0][i]=0;
        if(!c0) for(int i=0;i<n;i++) matrix[i][0]=0;

        //return matrix;



    }
};
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38