欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

矩阵置零

程序员文章站 2022-07-15 12:23:52
...

题目

来源:力扣(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;
            }
        }
    }

};
相关标签: LeetCode刷题记录