编写一种算法,若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]
]
思路:
标记零的位置,对零的位置的该行该列进行值的修改,修改为零。
代码:
class Solution {
public:
void changeRow(vector<vector<int>>& matrix,int row)//修改行
{
for(int c=0;c<matrix[0].size();c++)
{
matrix[row][c]=0;
}
}
void changeCol(vector<vector<int>>& matrix,int col)//修改列
{
for(int r=0;r<matrix.size();r++)
{
matrix[r][col]=0;
}
}
void setZeroes(vector<vector<int>>& matrix) {
vector<vector<int>> zero;
int row = matrix.size();
int col = matrix[0].size();
for(int r=0;r<row;r++)//遍历数组,寻找零的位置
{
for(int c=0;c<col;c++)
{
if(matrix[r][c]==0)
{
zero.push_back({r,c});
}
}
}
for(int i=0;i<zero.size();i++)//修改每行每列
{
int zr = zero[i][0];
int zc = zero[i][1];
changeRow(matrix,zr);
changeCol(matrix,zc);
}
}
};
运行结果: