给定一个 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; } } } };