矩阵归零

代码：

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截图：