https://leetcode-cn.com/problems/set-matrix-zeroes/
思路:用一个变量记录第一列是否有0,然后就可以把数组的第一行和第一列作为标记位了。如果
m
[
i
]
[
j
]
=
0
m[i][j]=0
m[i][j]=0,就令
m
[
i
]
[
0
]
=
m
[
0
]
[
j
]
=
0
m[i][0]=m[0][j]=0
m[i][0]=m[0][j]=0,标记第
i
i
i行、第
i
i
i列需要置为零。最后更新的时候要倒着来或者从第二行开始,因为第一行被用作标记位了,在处理完其它数据之前不能更新,不然可能会有问题,举例:
0 2 0
1 2 3
1 2 3
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
int n=matrix.size(),m=matrix[0].size();
bool flag=false;
for(int i=0;i<n;i++)
{
if(!matrix[i][0])
flag=true;
for(int j=1;j<m;j++)
if(!matrix[i][j])
matrix[i][0]=matrix[0][j]=0;
}
for(int i=n-1;i>=0;i--)
{
for(int j=1;j<m;j++)
if(!matrix[i][0]||!matrix[0][j])
matrix[i][j]=0;
if(flag)
matrix[i][0]=0;
}
}
};