C++小白刷题Leetcode【二维数组-零矩阵】
提示:以下是本篇文章正文内容,下面案例可供参考
一、题目
编写一种算法,若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]
]
二、思路
一开始我想的是:由于题目要求只要是有“0”的行和列都赋值为“0”,那我就首先找等于“0”的数组值然后将其对应的行和列赋值为“0”,然后依次再继续往下找为“0”的点。然后发现这样下去最后整个数组都会为0。因为例如
[0,1,2,0],
[3,4,5,2],
[1,3,1,5]
依照我一开始的思路,从头到尾的遍历到发现[0][0]的值为0,那么此时将它的行和列都为0就变成
[0,0,0,0],
[0,4,5,2],
[0,3,1,5]
那当到[0][1]的时候发现此时原本的“1”变成了"0"那么也继续按照规则将[0][1]所对应的行和列赋值为0.依次下去整个数组就就全为0了。所以否绝!!
## 正确代码
正确思路如下():
既然只要是有“0”的那一排,那一列都变为0,那我们应该在一开始就找出其值为“0”的位置坐标,然后将它的横,纵坐标使用特殊相应的标记记录下来
(我这里使用的bool类型的true),让它只要发现被标记了ture的横或者纵坐标都变为“0”即可
```c
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
int n = matrix.size(); //数组的行的size。
int c = matrix[0].size();//数组的列的size。
vector<bool> row(n) , col(c); //创建2个一维数组用来记录被标记了ture的横或者纵坐标。
int i,j;
for(i=0;i<n;i++)
{
for(j=0;j<c;j++)
{
if(matrix[i][j]==0)
{
row[i]=true;//标记了ture的横坐标
col[j]=true;//标记了ture的纵坐标
}
}
}
for(i=0;i<n;i++)
{
for(j=0;j<c;j++)
{
if( row[i]==true||col[j]==true)//只要发现被标记了ture的横或者纵坐标都变为“0”
{
matrix[i][j]=0;
}
}
}
}
};
总结
以上就是今天要讲的内容,本小白也是刚开始接触Leetcode来刷题,有更好的思路请多多交流,谢谢。