矩陣置零

題目

來源：力扣（LeetCode）
鏈接：矩陣置零

代碼

空間複雜度爲o(m+n)

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        set<int> rows;//保存含0的行下標
        set<int> columns;//保存含0的列下標
        for(int i = 0; i < matrix.size(); i++){
            for(int j = 0; j < matrix[i].size(); j++){
                if(matrix[i][j] == 0){
                    rows.insert(i);
                    columns.insert(j);
                }
            }
        }
        for(auto index : columns){
            for(int i = 0; i < matrix.size(); i++){
                matrix[i][index] = 0;
            }
        }
        for(auto index : rows){
            for(int j = 0; j < matrix[index].size(); j++){
                matrix[index][j] = 0;
            }
        }
    }

};