Eric 的小站

1
输入：matrix = [[1,1,1],[1,0,1],[1,1,1]]
2
输出：[[1,0,1],[0,0,0],[1,0,1]]

1
输入：matrix = [[0,1,2,0],[3,4,5,2],[1,3,1,5]]
2
输出：[[0,0,0,0],[0,4,5,0],[0,3,1,0]]

1
class Solution {
2
public:
3
    void setZeroes(vector<vector<int>>& matrix) {
4
        unordered_map<int, int> iMap;
5
        unordered_map<int, int> jMap;
6
        for (int i = 0; i < matrix.size(); i++) {
7
            for (int j = 0; j < matrix[0].size(); j++) {
8
                if (matrix[i][j] == 0) {
9
                    iMap[i]++;
10
                    jMap[j]++;
11
                }
12
            }
13
        }
14
        for (int i = 0; i < matrix.size(); i++) {
15
            for (int j = 0; j < matrix[0].size(); j++) {
16
                if (iMap.find(i) != iMap.end()) {
17
                    matrix[i][j] = 0;
18
                } else if (jMap.find(j) != jMap.end()) {
19
                    matrix[i][j] = 0;
20
                }
21
            }
22
        }
23
    }
24
};