【leetcode】73.(Medium)Set Matrix Zeroes
阿新 • • 發佈:2018-11-22
解題思路:
首先空間複雜度為O(m*n)的做法是,直接copy一套matrix來記錄
空間複雜度為O(m+n)的做法是設定陣列row[m]和column[n]來記錄每行每列的0的情況
空間複雜度為O(1)的情況就是設定兩個boolean變數,首先判定第一行和第一列是否會最終變為0,然後從第一行第一列(最左上角位置為第0行第0列)開始,更新matrix。最後再根據這兩個boolean變數來更新matrix的第一行和第一列。
提交程式碼:
class Solution {
public void setZeroes(int[][] matrix) {
boolean zeroColumn= false,zeroRow=false;
for(int i=0;i<matrix.length;i++)
if(matrix[i][0]==0) {
zeroColumn=true;
break;
}
for(int j=0;j<matrix[0].length;j++)
if(matrix[0][j]==0) {
zeroRow=true;
break;
}
for(int i=0;i< matrix.length;i++)
for(int j=0;j<matrix[0].length;j++)
if(matrix[i][j]==0) {
matrix[i][0]=0;
matrix[0][j]=0;
}
for(int i=1;i<matrix.length;i++)
for(int j=1;j<matrix[0].length;j++)
if(matrix[i][0]==0||matrix[0][j]==0)
matrix[i][j]=0;
if (zeroColumn)
for(int i=0;i<matrix.length;i++)
matrix[i][0]=0;
if(zeroRow)
for(int j=0;j<matrix[0].length;j++)
matrix[0][j]=0;
}
}
執行結果: