矩阵置零
leecode一道题目:
给定一个 m x n 的矩阵,如果一个元素为 0,则将其所在行和列的所有元素都设为 0。请使用原地算法
输入: [ [1,1,1], [1,0,1], [1,1,1] ] 输出: [ [1,0,1], [0,0,0], [1,0,1] ]
进阶:
- 一个直接的解决方案是使用 O(mn) 的额外空间,但这并不是一个好的解决方案。
- 一个简单的改进方案是使用 O(m + n) 的额外空间,但这仍然不是最好的解决方案。
- 你能想出一个常数空间的解决方案?
直接的想法是用哈希表存储需要置零的行列。
class Solution { public: void setZeroes(vector<vector<int>>& matrix) { unordered_map<int, int> rows; unordered_map<int, int> cols; for(int i = 0; i < matrix.size(); i++) { for(int j = 0; j < matrix[i].size(); j++) { if(matrix[i][j] == 0) { rows.insert(make_pair(i, 0)); cols.insert(make_pair(j, 0)); } } } for(auto &p : rows) { int row = p.first; for(int i = 0; i < matrix[0].size(); i++) { matrix[row][i] = 0; } } for(auto &p : cols) { int col = p.first; for(int i = 0; i < matrix.size(); i++) { matrix[i][col] = 0; } } } };
The Safest Way to Get what you Want is to Try and Deserve What you Want.