其他分享
首页 > 其他分享> > 73. 矩阵置零

73. 矩阵置零

作者:互联网

给定一个 m x n 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用原地算法。

  示例 1:

      

 

 


    输入:matrix = [[1,1,1],[1,0,1],[1,1,1]]
    输出:[[1,0,1],[0,0,0],[1,0,1]]

===================================================================

思路:遍历1遍,记录行和列的位置

上代码:

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        unordered_set<int> h;
        unordered_set<int> l;
        for (int i = 0; i < matrix.size(); i++) {
            for (int j = 0; j < matrix[i].size(); j++) {
                if (matrix[i][j] == 0) {
                    h.emplace(i);
                    l.emplace(j);
                }
                    
            }
            
        }
        for (auto a : h) {
            for (int i = 0; i < matrix[a].size(); i++) {
                matrix[a][i] = 0;
            }
        }
        for (auto a : l) {
            for (int i = 0; i < matrix[i].size(); i++) {
                matrix[i][a] = 0;
            }
        }
    }
};

 

标签:set,matrix,int,矩阵,++,73,unordered,size
来源: https://www.cnblogs.com/4jiu/p/15788677.html