矩阵置零
给定一个 m x n 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。
示例 1:
输入:matrix = [[1,1,1],[1,0,1],[1,1,1]]
输出:[[1,0,1],[0,0,0],[1,0,1]]
示例 2:
输入:matrix = [[0,1,2,0],[3,4,5,2],[1,3,1,5]]
输出:[[0,0,0,0],[0,4,5,0],[0,3,1,0]]
提示:
m == matrix.length
n == matrix[0].length
1 <= m, n <= 200
-231 <= matrix[i][j] <= 231 - 1
作者:力扣 (LeetCode)
链接:https://leetcode-cn.com/leetbook/read/top-interview-questions-medium/xvmy42/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
这道题首先的想法是,遍历这个二维vector容器,得到其0位置的索引,然后将对应位置都记为0.
int n = matrix[0].size(); //列
int m = matrix.size(); //行
vector<int> row(m),col(n);
for(int i = 0;i<n;i++){
for(int j = 0;j<m;j++){
if(matrix[i][j] == 0){
row[i]=1;
col[j]=1;
}
}
}
在这个代码里,我们声明了两个一维容器,vector,分别作为记录matrix的行和列,当matrix中的元素为0时,我们给row和col分别赋值为1.为什么这样赋值呢,因为这道题是让你将原matrix的所有0的行和列值都赋值为0.
for(int i = 0;i<n;i++){
for(int j = 0;j<m;j++){
if(row[i] == 1 || col[j] == 1){
matrix[i][j] = 0;
}
}
}
我们以例一为例,也就是row[1] = 1 ,col[1] = 1;matrix[1][1]=0;
我们需要的是将matrix[0][1]=0,matrix[1][1]=0,matrix[2][1] = 0;matrix[1][0] = 0,matrix[1][1] = 0 matrix[1][2] = 0;