一、题目
编写一种算法,若M × N矩阵中某个元素为0,则将其所在的行与列清零。
例1:
输入:
[
[1,1,1],
[1,0,1],
[1,1,1]
]
输出:
[
[1,0,1],
[0,0,0],
[1,0,1]
]
例2:
输入:
[
[0,1,2,0],
[3,4,5,2],
[1,3,1,5]
]
输出:
[
[0,0,0,0],
[0,4,5,0],
[0,3,1,0]
]
二、题目理解
1、遍历查找零元素,其对应的行数和列数,令该行所有元素为零,该列所有元素为零。
auto newm = m; % 建立一个新的数组
for(int i=0; i<n; i++){
for(int j=0; j<n; j++){
if(m[i][j]==0)
{
newm[n-1-i][0] = i; % 对应的行存储在newm数组的第一行第一列;
newm[n-1-i][1] = j; % 对应的列存储在newm数组的第一行第二列;
}
}
}
for(int ii=0; ii<newm.size(); ii++){
for(int jj=0; jj<2; jj++){
if(newm[ii][jj]!=0){
m[ii][所有列] = 0;
m[所有行][jj] = 0;
}
}
}
结果不对!!
针对第一个例子:
[
[1,1,1],
[1,0,1],
[1,1,1]
]
简单写一下遍历过程。
当i = 0, j = 0; m[0][0]=1; 不为零;
当i = 0; j = 1; m[0][1]=1; 不为零
当i = 0; j = 2; m[0][2]=1; 不为零
当i = 1, j = 0; m[1][0]=1; 不为零;
当i = 1; j = 1; m[1][1]=0; 为零
;
当i = 1; j = 2; m[1][2]=1; 不为零
当i = 2, j = 0; m[2][0]=1; 不为零;
当i = 2; j = 1; m[2][1]=1; 不为零
当i = 2; j = 2; m[2][2]=1; 不为零
当i = 1; j = 1; m[1][1]=0; 为零
;
令[1][0] , [1][2] 还有[2][1], [0][1]
为零。
用程序表示为 A[1][j] = 0 和A[i][1] =0;
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
int row = matrix.size(); // 数组的行数
int col = matrix[0].size(); // 数组的列数
if(row==0 && col==0) return; // 若为空矩阵,返回空值
int flag_row = 0; // 行标记:是否为0元素
int flag_col = 0; // 列标记,是否有0元素
//先扫描第一行第一列,如果有0,则将各自的flag设置为true
for(int i=0; i<row; i++){
if(matrix[i][0] == 0){
flag_col = 1;
}
}
for(int i=0; i<col; i++){
if(matrix[0][i] == 0){
flag_row = 1;
}
}
//扫描除去第一行第一列的整个数组,如果有0,则将对应的第一行和第一列的数字赋0
for(int i=0; i<row; i++){
for(int j=0; j<col; j++){
if(matrix[i][j] == 0){
matrix[0][j]=0;
matrix[i][0]=0;
}
}
}
// 遍历除去第一行第一列的整个数组,如果对应的第一行和第一列的数字有一个为0,则将当前值赋0
for(int i=0; i<row; i++){
for(int j=0; j<col; j++){
if(matrix[i][0] == 0 || matrix[0][j] == 0){
matrix[i][j]=0;
}
}
}
// 根据第一行第一列的flag来更新第一行第一列
if(flag_row){// 若是改该行存在零元素,那就遍历所有的列,令其为0
for(int i=0; i<row; i++){
matrix[0][i] = 0;
}
}
if(flag_col){// 若是改该列存在零元素,那就遍历所有的行,令其为0
for(int j=0; j<col; j++){
matrix[j][0]=0;
}
}
}
};
提交成功了!!!
三、他山之石
1、官方答案
感觉没看懂今天早上的题目,两个测试用例都是正确的,没得说!!!!
2、便于理解
利用队列的思想去做,好像更容易理解!