2021.11.3 接雨水

class Solution:

    def trapRainWater(self, heightMap: List[List[int]]) -> int:

        if len(heightMap) <= 2 or len(heightMap[0]) <= 2:

            return 0

        m, n = len(heightMap), len(heightMap[0])

        visited = [[0 for _ in range(n)] for _ in range(m)]

        pq = []

        for i in range(m):

            for j in range(n):

                if i == 0 or i == m - 1 or j == 0 or j == n - 1:

                    visited[i][j] = 1

                    heapq.heappush(pq, (heightMap[i][j], i * n + j))

        

        res = 0

        dirs = [-1, 0, 1, 0, -1]

        while pq:

            height, position = heapq.heappop(pq)

            for k in range(4):

                nx, ny = position // n + dirs[k], position % n + dirs[k + 1]

                if nx >= 0 and nx < m and ny >= 0 and ny < n and visited[nx][ny] == 0:

                    if height > heightMap[nx][ny]:

                        res += height - heightMap[nx][ny]

                    visited[nx][ny] = 1    

                    heapq.heappush(pq, (max(height, heightMap[nx][ny]), nx * n + ny))

        return res

标签：heightMap,pq,2021.11,雨水,height,nx,ny,range
来源： https://blog.csdn.net/q535030343/article/details/121132297