矩阵置零

题目

来源：力扣（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;
            }
        }
    }

};