力扣刷题学习407. 接雨水 II（C++）

题目描述

直接上原链接看题吧

解答

首先可以明确能接雨水的方块特点

• 不能是最外层方块
• 周围一圈方块高度都比它高

这里我们假设逐个方块注水，那么要从外往里逐个检查，因为最外层最低的方块决定了当前圈层的最低高度。
首先我们每次遍历一遍当前最外层的方块，选择当前圈中最低的方块检查其周围的方块是否是访问过的，
如果未访问过且高度比其低，则将其注水，且更新高度为注水后的高度，如此遍历完整个数组。

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)。