LeetCode第73题矩阵置零
1.思路: 想到一个开辟一点空间来解决方法,使用哈希集。就是使用一个哈希集(row和col)来储存数组中的元素为0的下标。然后再遍历,整个二维数组,在哈希集中存在对应的下标,就将这一行或这一列置为0。
2.代码部分
cpp
#include<iostream>
#include<vector>
#include <unordered_set>
using namespace std;
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
unordered_set<int> row; //记录横坐标
unordered_set<int> col; //记录纵坐标
for (int i = 0; i < matrix.size(); i++) {
for (int j = 0; j < matrix[0].size(); j++) {
if (matrix[i][j] == 0) {
row.insert(i);
col.insert(j);
}
}
}
for (int i = 0; i < matrix.size(); i++) {
for (int j = 0; j < matrix[0].size(); j++) {
if (row.count(i) || col.count(j)) {
matrix[i][j] = 0;
}
}
}
}
};