1. 程式人生 > 實用技巧 >leetcode 76:set-matrix-zeros

leetcode 76:set-matrix-zeros

題目描述

給定一個m*n的矩陣,如果有一個元素是0,就把該元素所在的行和列上的元素全置為0,要求使用原地演算法。 拓展: 你的演算法有使用額外的空間嗎? 一種比較直接的演算法是利用O(m,n)的空間,但是這不是一個好的解法 使用簡單的改進可以在O(m+n)的空間解決這個問題,但是還不是最佳的解法 你能在常量級的空間複雜度內解決這個問題嗎? 題目分析: 建立兩個map分別記載矩陣中值為0的行和列(map可去重),然後再遍歷這兩個map,如果有行和列被標記則把該行該列的元素全置為0。 程式碼如下:
 1 void setZeroes(vector<vector<int> > &matrix) {
2 int m = matrix.size(); 3 int n = matrix[0].size(); 4 map<int,bool> row,col; 5 for(int i = 0;i < m;i++) 6 { 7 for(int j = 0;j < n;j++) 8 { 9 if(matrix[i][j] == 0) 10 row[i] = col[j] = true
; 11 } 12 } 13 14 for(int i = 0;i < m;i++) 15 { 16 for(int j = 0;j < n;j++) 17 { 18 if(row[i] || col[j]) 19 matrix[i][j] = 0; 20 } 21 } 22 }