题目描述
解答
首先可以明确能接雨水的方块特点
- 不能是最外层方块
- 周围一圈方块高度都比它高
这里我们假设逐个方块注水,那么要从外往里逐个检查,因为最外层最低的方块决定了当前圈层的最低高度。
首先我们每次遍历一遍当前最外层的方块,选择当前圈中最低的方块检查其周围的方块是否是访问过的,
如果未访问过且高度比其低,则将其注水,且更新高度为注水后的高度,如此遍历完整个数组。
typedef pair<int,int> pii;
class Solution {
public:
int trapRainWater(vector<vector<int>>& heightMap) {
if (heightMap.size() <= 2 || heightMap[0].size() <= 2) {
return 0;
}
int m = heightMap.size();
int n = heightMap[0].size();
priority_queue<pii, vector<pii>, greater<pii>> pq;
vector<vector<bool>> visit(m, vector<bool>(n, false));
for (int i = 0; i < m; ++i) {
for (int j = 0; j < n; ++j) {
if (i == 0 || i == m - 1 || j == 0 || j == n - 1) {
pq.push({heightMap[i][j], i * n + j});
visit[i][j] = true;
}
}
}
int res = 0;
int dirs[] = {-1, 0, 1, 0, -1};
while (!pq.empty()) {
pii curr = pq.top();
pq.pop();
for (int k = 0; k < 4; ++k) {
int nx = curr.second / n + dirs[k];
int ny = curr.second % n + dirs[k + 1];
if( nx >= 0 && nx < m && ny >= 0 && ny < n && !visit[nx][ny]) {
if (heightMap[nx][ny] < curr.first) {
res += curr.first - heightMap[nx][ny];
}
visit[nx][ny] = true;
pq.push({max(heightMap[nx][ny], curr.first), nx * n + ny});
}
}
}
return res;
}
};
复杂度分析
- 时间复杂度:O(MN\log(M+N))O(MNlog(M+N)),其中 MM 是矩阵的行数,NN 是矩阵的列数。我们需要将矩阵中的每个元素都进行遍历,同时将每个元素都需要插入到优先队列中,总共需要向队列中插入 MNMN 个元素,每次堆进行调整的时间复杂度为 O(\log(M+N))O(log(M+N)),因此总的时间复杂度为 O(MN\log(M+N))O(MNlog(M+N))。
- 空间复杂度:O(MN)O(MN),其中 MM 是矩阵的行数,NN 是矩阵的列数。我们需要创建额外的空间对元素进行标记,优先队列中最多存储 O(MN)O(MN) 个元素,因此空间复杂度度为 O(MN)O(MN)。