開辟數組來標記元素為0的行和列,然后將對應的行和列的元素全部置為0;
class Solution {
public:void setZeroes(vector<vector<int>>& matrix) {int n = matrix.size();int m = matrix[0].size();vector<int> l(m),r(n);for(int i = 0;i<n;i++){for(int j =0;j<m;j++){if(matrix[i][j]==0) l[j]=r[i]=1;}}for(int i = 0;i<n;i++){for(int j =0;j<m;j++){if(r[i]||l[j])matrix[i][j]=0;} }}
};
時間復雜度:因為用了兩層for循環所以是O(mn)
空間復雜度:開辟了兩個數組,是O(m+n)