407. 接雨水 II
给你一个 m x n 的矩阵,其中的值均为非负整数,代表二维高度图每个单元的高度,请计算图中形状最多能接多少体积的雨水。
示例:
给出如下 3x6 的高度图:
[
[1,4,3,1,3,2],
[3,2,1,3,2,4],
[2,3,3,2,3,1]
]
返回 4 。
如上图所示,这是下雨前的高度图[[1,4,3,1,3,2],[3,2,1,3,2,4],[2,3,3,2,3,1]] 的状态。
下雨后,雨水将会被存储在这些方块中。总的接雨水量是4。
提示:
- 1 <= m, n <= 110
- 0 <= heightMap[i][j] <= 20000
解题思路: 这个要跟42. 接雨水对比着看,42题是在二维平面上求积水量,此题是在三维立体中求积水量,42题形成水坑的条件是左右高中间低,而此题形成水坑的条件是四周高中间低,42题解水坑高度时用了一个方法,从左往右遍历求左侧最大高度,从右往左遍历求右侧最大高度,当左右侧最大高度求得后水坑的高度取决于左右侧高度的较小者。那么我们分析此题,我们也可以维护一个水坑四周的最大值的较小者,在这里把它形象的称作上升中的水位,我们维护一个优先级队列,让水位高度由小变大,然后用BFS解题,将边界点push至队列作为BFS启动点,当遇见水位比访问点高时,说明这里可以形成水坑,即可求水坑的高度。
解完此题后,我心里还有个顾虑,访问到这个节点,若是水坑,立马求出的水坑高度一定是水坑的高度吗,因为水位在逐渐上上,后面上升到更大值的水位,会不会又进一步抬高了水坑的高度呢?因为我们水位是按照由低到高进行上升 访问的,在我们判定当前位置为水坑时,水位已经升高到能灌满它的位置了,水位再升高,也不能再往水坑里灌水。
class Solution {
public:
int trapRainWater(vector<vector<int>>& heightMap) {
if (heightMap.empty()) return 0;
priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> q;
int rows = heightMap.size(), cols = heightMap[0].size();
vector<vector<bool>> visited(rows, vector<bool>(cols, false));
for (int i = 0; i < rows; ++i) {
for (int j = 0; j < cols; ++j) {
if (i == 0 || i == rows - 1 || j == 0 || j == cols - 1) {
q.push({heightMap[i][j], i * cols + j});
visited[i][j] = true;
}
}
}
int mx = INT_MIN;
int res = 0;
vector<vector<int>> dirs{{0,-1},{-1,0},{0,1},{1,0}};
while (!q.empty()) {
auto t = q.top(); q.pop();
int h = t.first;
int r = t.second / cols, c = t.second % cols;
mx = max(mx, h);
for (auto dir : dirs) {
int x = r + dir[0], y = c + dir[1];
if (x < 0 || x >= rows || y < 0 || y >= cols || visited[x][y]) continue;
visited[x][y] = true;
if (mx > heightMap[x][y]) res += mx - heightMap[x][y];
q.push({heightMap[x][y], x * cols + y});
}
}
return res;
}
};
https://www.cnblogs.com/grandyang/p/5928987.html