清除行列
题目描述
请编写一个算法,若MxN矩阵中某个元素为0,则将其所在的行与列清零。
给定一个MxN的int[][]矩阵(C++中为vector<vector>)mat和矩阵的阶数n,请返回完成操作后的int[][]矩阵(C++中为vector<vector>),保证n小于等于300,矩阵中的元素为int范围内。
测试样例:
[[1,2,3],[0,1,2],[0,0,1]]
返回:[[0,0,3],[0,0,0],[0,0,0]]
思路:没想到比O(m*n)时间复杂度还快的,有思路的可以交流下;
没什么别的方法,我用row,line来记录行列中是否有0;首先初始化为1;当mat[i][j]出现0的时候,将row[i]=0;line[j]=0; 遍历一遍;
最后,再遍历一遍把row或者line里有0 的地方就把值变成0,否则记录原值!
方法比较直接。。。
/**
题目描述
请编写一个算法,若MxN矩阵中某个元素为0,则将其所在的行与列清零。
给定一个MxN的int[][]矩阵(C++中为vector>)mat和矩阵的阶数n,请返回
完成操作后的int[][]矩阵(C++中为vector>),保证n小于等于300,矩阵中的元素为int范围内。
测试样例:
[[1,2,3],
[0,1,2],
[0,0,1]]
返回: [[0,0,3],
[0,0,0],
[0,0,0]]
**/
#include<iostream>
#include<vector>
using namespace std;
class Clearer {
public:
vector<vector<int> > clearZero(vector<vector<int> > mat, int n) {
vector<vector<int> > vec;
if(mat.size()<=0||n==0) return vec;
vector<int> row(mat.size(),1);
vector<int> line(n,1);
for(int i=0;i<mat.size();i++)
{
for(int j=0;j<mat[i].size();j++)
{
if(!mat[i][j])
{
row[i]=0;
line[j]=0;
}
}
}
for(int i=0;i<mat.size();i++)
{
vector<int> tmp;
for(int j=0;j<mat[i].size();j++)
{
if(!row[i] || !line[j])
tmp.push_back(0);
else tmp.push_back(mat[i][j]);
}
vec.push_back(tmp);
tmp.clear();
}
return vec;
}
};
void Print(vector<vector<int> > &v)
{
for(size_t i=0;i<v.size();i++)
{
for(size_t j=0;j<v[i].size();j++)
cout<< v[i][j] <<" ";
cout<<endl;
}
}
int main()
{
vector<vector<int> >arr={{1,2,3},{0,1,2},{0,0,1}};
Print(arr);
Clearer cl ;
arr=cl.clearZero(arr,3);
Print(arr);
return 0;
}