赞
踩
1.借助两个一维数组r[rows],c[cols],r[rows]记录其对应行是否含有0,c[cols]记录其对应列是否含有0
2.遍历二维数组,如果matrix[i][j]==0,则r[i]=0,c[j]=0
3.遍历二维数组,如果r[i]==0||c[j]==0,则matrix[i][j]=0
class Solution { public: void setZeroes(vector<vector<int>>& matrix) { int rows=matrix.size(); int cols=matrix[0].size(); vector<int> r(rows,1); vector<int> c(cols,1); for(int i=0;i<rows;i++) { for(int j=0;j<cols;j++) { if(matrix[i][j]==0) { r[i]=0; c[j]=0; } } } for(int i=0;i<rows;i++) { for(int j=0;j<cols;j++) { if(r[i]==0||c[j]==0) { matrix[i][j]=0; } } } } };
1.无需借组额外数组记录每行和每列中是否含有0,只需要借组matrix[0][y]这一行和matrix[x][0]这一列记录每行和每列是否含有0
2.同1,遍历二维数组,如果matrix[i][j]==0,则matrix[i][0]=0,matrix[0][j]=0
3.遍历二维数组,如果r[i]==0||c[j]==0,则matrix[i][j]=0
3.对于用于记录的matrix[0][y]和matrix[x][0]这一行和这一列,如果matrix[0][j]==0,可能是原先就为0,也可能是在遍历过程中因为同一列某个元素为0导致其更新为0;对于情况1,我们事先用一个r来第一行记录是否含有0,如果含有0的话,我们只需要将第一行所有元素更新为0即可;第一列情况类似。
class Solution { public: void setZeroes(vector<vector<int>>& matrix) { int rows=matrix.size(); int cols=matrix[0].size(); int r=0; int c=0; for(int i=0;i<rows;i++) { if(matrix[i][0]==0) { r=1; break; } } for(int j=0;j<cols;j++) { if(matrix[0][j]==0) { c=1; break; } } for(int i=1;i<rows;i++) { for(int j=1;j<cols;j++) { if(matrix[i][j]==0) { matrix[0][j]=0; matrix[i][0]=0; } } } for(int i=1;i<rows;i++) { for(int j=1;j<cols;j++) { if(matrix[0][j]==0||matrix[i][0]==0) { matrix[i][j]=0; } } } if(r==1) { for(int i=0;i<rows;i++) matrix[i][0]=0; } if(c==1) { for(int j=0;j<cols;j++) matrix[0][j]=0; } } };
1.基本思路同2,只不过无需记录第一列是否含有0,思路二中,matrix[i][j]的内层更新遍历是从下标1–下标cols-1,然而这里内层遍历下标从cols-1到0,这样做就能够确保matrix[x][0]不被过早更新,保证matrix[x][0]在matrix[i][j]更新过程中,达到指示该行是否更新全0的效果。
2.最后还需要判断第一行是否含有0元素,如果有则更新第一行所有元素。
class Solution { public: void setZeroes(vector<vector<int>>& matrix) { int rows=matrix.size(); int cols=matrix[0].size(); int c=0; for(int j=0;j<cols;j++) { if(matrix[0][j]==0) { c=1; break; } } for(int i=1;i<rows;i++) { for(int j=0;j<cols;j++) { if(matrix[i][j]==0) { matrix[0][j]=0; matrix[i][0]=0; } } } for(int i=1;i<rows;i++) { for(int j=cols-1;j>=0;j--) { if(matrix[0][j]==0||matrix[i][0]==0) { matrix[i][j]=0; } } } if(c==1) { for(int j=0;j<cols;j++) matrix[0][j]=0; } } };
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。