leetcode 73 矩阵置零
给定一个 m x n 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。
进阶:
一个直观的解决方案是使用 O(mn) 的额外空间,但这并不是一个好的解决方案。
一个简单的改进方案是使用 O(m + n) 的额外空间,但这仍然不是最好的解决方案。
你能想出一个仅使用常量空间的解决方案吗?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/set-matrix-zeroes
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
举例:
0120 0000
?3452 ----》 0450
?0310 0310
法一,思路混乱乱写的
想着把每一个0的坐标记下来,之后再对每一个0处理,为了记坐标,想到了Pair
class Solution {
public void setZeroes(int[][] matrix) {
int m=matrix.length;
int n=matrix[0].length;
int [] row = new int[n];
//这里想着初始话一行0,之后将行置零的时候,可以直接克隆
for(int i=0;i<n;i++){
row[i]=0;
}
ArrayList<Pair<Integer,Integer>> arrayList = new ArrayList<>();
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
if(matrix[i][j]==0){
arrayList.add(new Pair(i,j));
}
}
}
for(Pair<Integer,Integer> p:arrayList){
matrix[p.getKey()] = row.clone();
for(int i=0;i<m;i++){
matrix[i][p.getValue()]=0;
}
}
}
}
上面这个方法,空间使用直接达到了O(m+n+m)(好像是吧,反正不小)而且解法挺蠢的
法二 使用两个标记变量
这个方法是看了题解后做出来的,还有引入一个标记变量的做法,我没看。
class Solution {
public void setZeroes(int[][] matrix) {
int m=matrix.length;
int n=matrix[0].length;
boolean row0=false;
boolean col0=false;
//第一列有没有0
for(int i=0;i<m;i++){
if(matrix[i][0]==0){
col0=true;
}
}
//第一行有没有0
for(int j=0;j<n;j++){
if(matrix[0][j]==0){
row0=true;
}
}
for(int i = 1;i < m; i++){
for(int j = 1;j < n; j++){
if(matrix[i][j]==0){
matrix[i][0]=matrix[0][j]=0;
}
}
}
//将第一行第一列外的元素遍历,如果该置零就置零
for(int i=1;i<m;i++){
for(int j=1;j<n;j++){
if(matrix[i][0]==0||matrix[0][j]==0){
matrix[i][j]=0;
}
}
}
//搞下第一行,第一列
if(row0){
for(int i=0;i<n;i++){
matrix[0][i]=0;
}
}
if(col0){
for(int i=0;i<m;i++){
matrix[i][0]=0;
}
}
}
}
这样空间使用就到了O(1)