CC07:清除行列
题目
请编写一个算法,若N阶方阵中某个元素为0,则将其所在的行与列清零。
给定一个N阶方阵int[]mat和矩阵的阶数n,请返回完成操作后的int[][]方阵(C++中为vector
测试样例:
[[1,2,3],[0,1,2],[0,0,1]]
返回:[[0,0,3],[0,0,0],[0,0,0]]
解法
这道题可以额外设置两个标志数组row与col,分别表示行与列,当出现0的时候就可以将相应的行与列row与col标志为1;最后重新遍历一次,如果行或者列的数组的标志为1,则将mat[row][col]置为0.
class Clearer {
public:
vector<vector<int> > clearZero(vector<vector<int> > mat, int n) {
// write code here
int c=mat.size();
int m=mat[0].size();
if(c<=0)
return mat;
bool *row=new bool[c]{false};
bool *col=new bool[m]{false};
for(int i=0;i<c;i++)
{
for(int j=0;j<m;j++)
{
if(mat[i][j]==0)
{
row[i]=true;
col[j]=true;
}
}
}
for(int i=0;i<c;i++)
{
for(int j=0;j<m;j++)
{
if(row[i] || col[j])
mat[i][j]=0;
}
}
delete []row;
delete []col;
return mat;
}
};
作者:YunLambert
-------------------------------------------
个性签名:一名会音乐、爱健身的不合格程序员
可以Follow博主的Github哦(っ•̀ω•́)っ✎⁾⁾