題目:
請編寫一個算法,若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;
}
};