LeetCode 73. Set Matrix Zeroes
题目描述:
Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in-place.
Example 1:
Input:
[
[1,1,1],
[1,0,1],
[1,1,1]
]
Output:
[
[1,0,1],
[0,0,0],
[1,0,1]
]
Example 2:
Input:
[
[0,1,2,0],
[3,4,5,2],
[1,3,1,5]
]
Output:
[
[0,0,0,0],
[0,4,5,0],
[0,3,1,0]
]
Follow up:
- A straight forward solution using O(mn) space is probably a bad idea.
- A simple improvement uses O(m + n) space, but still not the best solution.
- Could you devise a constant space solution?
题目理解:
将给定矩阵中0元素所在的行列都变成0.提示里面也说了,要求降低空间复杂度,暴力的直接申请一个M*N的空间就很舒服,或者申请两个vector记录为0的行和列这样复杂度就降低成了M+N,然后借鉴了网上优秀代码将空间复杂度降低成了1.
暴力美学:
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
int row = matrix.size();
int line = matrix[0].size();
vector<vector<int>> ans(matrix);
for(int i=0;i<row;i++){
for(int j=0;j<line;j++){
if(matrix[i][j] == 0){
for(int p=0;p<row;p++) ans[p][j] = 0;
for(int q=0;q<line;q++) ans[i][q] = 0;
}
}
}
matrix.swap(ans);
}
};
学习优秀代码:
思路理解:
- 在数组的第一列和第一行先找看看是否存在0,并将其标志存储起来;
- 然后遍历剩余数组,若发现0,则将对应位置的第一行和第一列元素变成0,这样因为之前第一行列是否有零的标志已经被记录了下来,所以这里不用担心破坏;
- 在找完0后,对除了第一行列外的剩余矩阵进行操作,用对应行列在第一行列的元素是否为0判断该行列是否都置0;
- 都完成后,再根据最开始存储的标志对第一行列,进行变0操作。
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
int row = matrix.size();
int line = matrix[0].size();
bool rflag,lflag;
rflag = lflag = false;
for(int i=0;i<row;i++){
if(!matrix[i][0]){//遍历第一列
lflag = true;
}
}
for(int j=0;j<line;j++){
if(!matrix[0][j]){
rflag = true;
}
}
for(int i=1;i<row;i++){
for(int j=1;j<line;j++){
if(matrix[i][j] == 0){
matrix[i][0] = 0;
matrix[0][j] = 0;
}
}
}
for(int i=1;i<row;i++){
for(int j=1;j<line;j++){
if(matrix[i][0] == 0 || matrix[0][j]==0){
matrix[i][j] = 0;
}
}
}
if(lflag){
for(int i=0;i<row;i++){
matrix[i][0] = 0;
}
}
if(rflag){
for(int j=0;j<line;j++){
matrix[0][j] = 0;
}
}
}
};
空间复杂度可以降低,时间复杂度无法操作。。。