Leetcode演算法Java全解答--73. 矩陣置零
阿新 • • 發佈:2018-12-26
Leetcode演算法Java全解答–73. 矩陣置零
文章目錄
題目
給定一個 m x n 的矩陣,如果一個元素為 0,則將其所在行和列的所有元素都設為 0。請使用原地演算法。
進階:
一個直接的解決方案是使用 O(mn) 的額外空間,但這並不是一個好的解決方案。
一個簡單的改進方案是使用 O(m + n) 的額外空間,但這仍然不是最好的解決方案。
你能想出一個常數空間的解決方案嗎?
示例 1:
輸入:
[
[1,1,1],
[1,0,1],
[1,1,1]
]
輸出:
[
[1,0,1],
[0,0,0],
[1,0,1]
]
示例 2:
輸入:
[
[0,1,2,0],
[3,4,5,2],
[1,3,1,5]
]
輸出:
[
[0,0,0,0],
[0,4,5,0],
[0,3,1,0]
]
想法
建立一個大小為 n 的陣列,用於記錄第 j 列是否出現0
結果
超過78%的測試案例
時間複雜度/空間複雜度:n/n
總結
程式碼
我的答案
public void setZeroes(int[][] matrix) { boolean[] flag = new boolean[matrix[0].length]; for(int i =0; i<matrix.length; i++) { boolean contains0 = false; for (int j =0; j < matrix[0].length; j++) { if(matrix[i][j] == 0) { contains0 = true; flag[j] = true; } } if(contains0) for(int j=0; j<matrix[0].length; j++) matrix[i][j] = 0; } for(int j = 0; j < matrix[0].length; j++) if(flag[j]) for(int i =0; i<matrix.length; i++) matrix[i][j] = 0; }
大佬們的答案
class Solution { /** 差不多就是暴力解 這題 time : O(n * m) space : O(1) */ public void setZeroes(int[][] matrix) { if (matrix == null || matrix.length == 0) return; int m = matrix.length; int n = matrix[0].length; boolean row = false; boolean col = false; for (int i = 0; i < m; i++) { for (int j = 0; j < n; j++) { if (matrix[i][j] == 0) { //將出現0位置的首行和首列的相應位置設定為0,用來設定內矩陣的0 matrix[0][j] = 0; matrix[i][0] = 0; if (i == 0) row = true;//當前位置本來就在首列, if (j == 0) col = true;//首行 } } } for (int i = 1; i < m; i++) { //遍歷行首為0的行,從第二行開始,所以後面需要用col的值來設定第一行 if (matrix[i][0] == 0) { for (int j = 1; j < n; j++) { matrix[i][j] = 0; } } } for (int j = 1; j < n; j++) { //遍歷列首為0的列,從第二列開始,所以後面需要用row的值來設定第一列 if (matrix[0][j] == 0) { for (int i = 1; i < m; i++) { matrix[i][j] = 0; } } } if (row) { for (int j = 0; j < n; j++) { matrix[0][j] = 0; } } if (col) { for (int i = 0; i < m; i++) { matrix[i][0] = 0; } } } }
測試用例
其他
程式碼託管碼雲地址:https://gitee.com/lizhaoandroid/LeetCodeAll.git
檢視其他內容可以點選專欄或者我的部落格哈:https://blog.csdn.net/cmqwan
“大佬們的答案” 標籤來自leetcode,侵權請聯絡我進行刪改
如有疑問請聯絡,聯絡方式:QQ3060507060