407. 接雨水 II

---

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

说明:

m 和 n 都是小于110的整数。每一个单位的高度都大于0 且小于 20000。

示例：

给出如下 3x6 的高度图:
[
  [1,4,3,1,3,2],
  [3,2,1,3,2,4],
  [2,3,3,2,3,1]
]

返回 4。

返回 4。

如上图所示，这是下雨前的高度图[[1,4,3,1,3,2],[3,2,1,3,2,4],[2,3,3,2,3,1]] 的状态。

下雨后，雨水将会被存储在这些方块中。总的接雨水量是4。

思路：参考https://blog.csdn.net/XX_123_1_RJ/article/details/81089495

# coding:utf-8
class Solution(object):
    def trapRainWater(self, heightMap):
        m = len(heightMap)
        n = len(heightMap[0]) if m else 0
        peakMap = [[float('inf')] * n for _ in range(m)]
        queue = []

        # 把矩形的四边入队
        for x in range(m):
            for y in range(n):
                if x in (0, m - 1) or y in (0, n -1):
                    peakMap[x][y] = heightMap[x][y]
                    queue.append((x, y))

        while queue:
            # 出队一个位置，并以这个位置更新四周的位置
            x, y = queue.pop(0)
            # 上下左右
            for dx, dy in zip((1, 0, -1, 0), (0, 1, 0, -1)):
                nx, ny = x + dx, y + dy

                # 不考虑边界值及超出范围的值
                if nx <= 0 or nx >= m - 1 or ny <= 0 or ny >= n - 1:
                    continue
                limit = max(peakMap[x][y], heightMap[nx][ny])
                # peakMap仅保存较大的值
                if peakMap[nx][ny] > limit:
                    peakMap[nx][ny] = limit
                    queue.append((nx, ny))

        return sum(peakMap[x][y] - heightMap[x][y] for x in range(m) for y in range(n))