[LeetCode] 407. 接雨水 II

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