首页 > 代码库 > 矩阵归零
矩阵归零
代码:
class Solution {
public:
/**
* @param matrix: A list of lists of integers
* @return: Void
*/
void setZeroes(vector<vector<int> > &matrix) {
// write your code here
int n = matrix.size();
if(n==0)
return ;
int m = matrix[0].size();
bool firstLine = false;
bool firstColumn = false;
for(int i=0;i<n;i++)
if(matrix[i][0]==0)
firstColumn=true;
for(int i=0;i<m;i++)
if(matrix[0][i]==0)
firstLine = true;
for(int i=1;i<n;i++)
{
for(int j=1;j<m;j++)
{
if(matrix[i][j]==0)
{
matrix[i][0]=0;
matrix[0][j]=0;
}
}
}
for(int i=1;i<n;i++)
{
if(matrix[i][0]==0)
{
for(int j=0;j<m;j++)
matrix[i][j]=0;
}
}
for(int i=1;i<m;i++)
{
if(matrix[0][i]==0)
{
for(int j=0;j<n;j++)
matrix[j][i]=0;
}
}
if(firstLine)
for(int i=0;i<m;i++)
matrix[0][i]=0;
if(firstColumn)
for(int i=0;i<n;i++)
matrix[i][0]=0;
}
};
lintcode截图:
矩阵归零