接雨水二维数组

1、描述

给你一个 m x n 的矩阵，其中的值均为非负整数，代表二维高度图每个单元的高度，请计算图中形状最多能接多少体积的雨水。
来源

2、关键字

接雨水
二维数组

3、思路

根据一维数组接雨水的思路，两边选两个边界，往里边填充，找到中间最大的，左边是一个单调不减数组，右边从右往左也是一个单调不减数组，左右两边同时往中间走，都到最高点就行了。

这个二维，就是把周围一圈全部搞到小顶堆，然后根据周围的元素去更新“边界”：
使用一个方向数组 dir[ 5 ]来遍历四周，
使用一个visit标记有没有北方问过，
更新的过程中，统计结果。

4、notes

腐烂的橘子
死亡细胞复活
陆地填海
接雨水1维
可以使用队列来初始化一圈，可以使用小根堆，大根堆，来初始化一圈。

5、复杂度

时间：O(MNlog(M+N))，插入操作M x N行数和列数，和调整log(M+N)
空间：O（N x M)

6、code

typedef pair<int,int> pii;  // 第一个元素放高度，第2个元素记录座标，通过 / % 获取横纵座标
class Solution {
public:
    int trapRainWater(vector<vector<int>>& heightMap) {
        int n = heightMap.size();
        int m = heightMap[0].size();
        if(n < 3 || m < 3) return 0;  // 没机会直接返回
        priority_queue<pii, vector<pii>, greater<pii>> pq;  // 构建小顶堆
        vector<vector<bool>>visit(n, vector<bool>(m, false)); // 标记初始化为 false
        for(int i = 0; i < n; i++){
            for(int j = 0; j < m; j++){
                if(i == 0 || i == n - 1 || j == 0 || j == m - 1){ // 把一圈边界进堆
                    pq.push({heightMap[i][j], i * m + j}); // 有一个push方法，
                    visit[i][j] = true;
                }
                
            }
        }
        int dir[5] = { 1, 0, -1, 0 ,1};
        int res = 0;
        while(!pq.empty()){
            auto cur = pq.top();
            pq.pop();
            for(int i = 0; i < 4; i++){
                int nx = cur.second / m + dir[i];  // 这里是dir[i], 而不是直接 i
                int ny = cur.second % m + dir[i + 1];
                if(nx > 0 && nx < n - 1 && ny > 0 && ny < m - 1 && !visit[nx][ny]){ //候选集中
                    if(heightMap[nx][ny] < cur.first){ // 当前值小于边界值，res + 差
                        res += cur.first - heightMap[nx][ny];
                    }
                    pq.push({max(heightMap[nx][ny], cur.first), nx * m + ny}); // 更新边界值
                    visit[nx][ny] = true;
                }

            }
        }
        return res;

    }
};