对于一个M*N的矩阵,可以很简单的想到,用一个大为M的数组标志某行是否有零,用一个大为N的数组标志某列是否有零
进而,我们用矩阵内的第一行和第一列作为标志数组。当某行第一列有零,flag=true
标志第一列有零(该算法会破坏第一列/第一行信息,需要记录其中之一是否有零),该行某列为零,则该行第一列为零,该列第一行为零。
最后把第一行/列为零的列/行,全部置零,若flag=true
则第一列全部置零。
时间复杂度:O(MN)
C++代码:
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
int m = matrix.size(), n = matrix[0].size();
bool flag = false;
for(int i = 0; i < m; i++)
{
if(matrix[i][0] == 0) flag = true;
for(int j = 1; j < n; j++)
{
if(matrix[i][j] == 0) matrix[i][0] = matrix[0][j] = 0;
}
}
for(int i = m-1; i >=0 ; i--)
{
for(int j = n-1; j > 0; j--)
{
if(matrix[i][0] == 0 || matrix[0][j] == 0) matrix[i][j] = 0;
}
if(flag) matrix[i][0] = 0;
}
}
};