矩阵置零

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;
            }   
        }
    }
};

 

转载于:https://www.cnblogs.com/Shinered/p/9726705.html