题目:
请编写一个算法,若MxN矩阵中某个元素为0,则将其所在的行与列清零。
给定一个MxN的int[][]矩阵(C++中为vector<vector>)mat和矩阵的阶数n,请返回完成操作后的int[][]矩阵(C++中为vector<vector>),保证n小于等于300,矩阵中的元素为int范围内。
思路:定义两个bool类型的数组,分别表示矩阵的行或列是否出现0。
第一步:扫描整个矩阵,如果检测到某个元素为0,则将该元素所在行和列对应bool数组置1。
第二步:扫描bool数组,然后将该元素对应的行或列的元素全部置为0。
实现:
class Clearer {
public:
vector<vector<int> > clearZero(vector<vector<int> > mat, int n) {
if(n==1) return mat;
vector<bool> hang(n,0);
vector<bool> lie(n,0);
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
if(mat[i][j]==0){
hang[i]=1;
lie[j]=1;
}
}
}
for(int i=0;i<n;i++){
if(hang[i]==1){
for(int j=0;j<n;j++){
mat[i][j]=0;
}
}
if(lie[i]==1){
for(int j=0;j<n;j++){
mat[j][i]=0;
}
}
}
return mat;
}
};