时间轴

2025-11-17

init

题目：

P73 矩阵置零

https://leetcode.cn/problems/set-matrix-zeroes/?envType=study-plan-v2&envId=top-interview-150

最开始我是这样写的，但是用哈希表存储效率很低。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
#include <vector>
#include <unordered_set>

using std::vector;
using std::unordered_set;

class Solution {
    public:
	void setZeroes(vector<vector<int> > &matrix)
	{
		unordered_set<int> iuset;
		unordered_set<int> juset;
		int i, j, m = matrix.size(), n = matrix[0].size();
		for (i = 0; i < m; i++) {
			for (j = 0; j < n; j++) {
				if (matrix[i][j] == 0) {
					iuset.insert(i);
					juset.insert(j);
				}
			}
		}
		for (i = 0; i < m; i++) {
			if (iuset.count(i)) { // 这一行都为0
				std::fill(matrix[i].begin(), matrix[i].end(), 0);
				continue;
			}
			for (j = 0; j < n; j++) {
				if (juset.count(j)) {
					matrix[i][j] = 0;
				}
			}
		}
	}
};

可以改成数组形式记录：
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int m = matrix.size();
        int n = matrix[0].size();
        vector<int> row(m), col(n);
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (!matrix[i][j]) {
                    row[i] = col[j] = true;
                }
            }
        }
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (row[i] || col[j]) {
                    matrix[i][j] = 0;
                }
            }
        }
    }
};

如果要再节省空间的话，可以找到数组中第一个为0的元素，用它的行列数组当作记录为0的行/列。最后再把它置零。