您的位置:首页 > 其它

leetcode:Set Matrix Zeors 菜鸟解法

2015-03-09 23:43 218 查看
这题本来想的挺好,觉得只要两个数组,一个记录哪行出现过0,一个记录哪列出现个0就好了,结果发现原来不能额外空间...

所以用第一行和第一列代替了,先查看第一列与第一行是否有0元素,之后可按之前的行列数组法对剩余数组进行处理。

代码如下:

class Solution {
public:
void setZeroes(vector<vector<int> > &matrix) {
int m = matrix.size();
int n = matrix[0].size();
int i=0,j=0,row=0,col=0;

for(;i<m;++i){
if(matrix[i][0]==0){
col = 1;
break;
}
}
for(;j<n;++j){
if(matrix[0][j]==0){
row = 1;
break;
}
}

for(i=1;i<m;++i){
for(j=1;j<n;++j){
if(matrix[i][j]==0){
matrix[0][j]=0;
matrix[i][0]=0;
}
}
}

for(i=1;i<m;++i){
for(j=1;j<n;++j){
if(matrix[i][0]==0||matrix[0][j]==0){
matrix[i][j]=0;
}
}
}

if(col){
for(i=0;i<m;++i){
matrix[i][0]=0;
}
}

if(row){
for(j=0;j<n;++j){
matrix[0][j]=0;
}
}

}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  leetcode matrix