1. 程式人生 > >【leetcode】73.(Medium)Set Matrix Zeroes

【leetcode】73.(Medium)Set Matrix Zeroes

解題思路:
首先空間複雜度為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; } }

執行結果:
在這裡插入圖片描述