1765. 地图中的最高点

Difficulty
Medium
Tags
BFS
图
URL
https://leetcode-cn.com/problems/map-of-highest-peak/
Star
给你一个大小为 m x n 的整数矩阵 isWater ，它代表了一个由 陆地 和 水域 单元格组成的地图。
  • 如果 isWater[i][j] == 0 ，格子 (i, j) 是一个 陆地 格子。
  • 如果 isWater[i][j] == 1 ，格子 (i, j) 是一个 水域 格子。
你需要按照如下规则给每个单元格安排高度：
  • 每个格子的高度都必须是非负的。
  • 如果一个格子是是 水域 ，那么它的高度必须为 0 。
  • 任意相邻的格子高度差 至多 为 1 。当两个格子在正东、南、西、北方向上相互紧挨着，就称它们为相邻的格子。（也就是说它们有一条公共边）
找到一种安排高度的方案，使得矩阵中的最高高度值 最大 。
请你返回一个大小为 m x n 的整数矩阵 height ，其中 height[i][j] 是格子 (i, j) 的高度。如果有多种解法，请返回 任意一个 。
示例 1：
notion imagenotion image
输入：isWater = [[0,1],[0,0]]
输出：[[1,0],[2,1]]
解释：上图展示了给各个格子安排的高度。
蓝色格子是水域格，绿色格子是陆地格。
示例 2：
输入：isWater = [[0,0,1],[1,0,0],[0,0,0]]
输出：[[1,1,0],[0,1,1],[1,2,2]]
解释：所有安排方案中，最高可行高度为 2 。
任意安排方案中，只要最高高度为 2 且符合上述规则的，都为可行方案。
提示：
  • m == isWater.length
  • n == isWater[i].length
  • 1 <= m, n <= 1000
  • isWater[i][j] 要么是 0 ，要么是 1 。
  • 至少有 1 个水域格子。

法 1 BFS

思路
类似
⌚
542. 01 矩阵 。本题目从 水池出发，将周围的节点值 令为 当前节点值 + 1.
题解
class Solution:
    def highestPeak(self, isWater: List[List[int]]) -> List[List[int]]:
        m, n = len(isWater), len(isWater[0])
        visited = [[False] * n for _ in range(m)]
        res = [[0] * n for _ in range(m)]
        queue = deque()
        dirs = ((-1, 0), (1, 0), (0, -1), (0, 1))
        
        # 从 水域出发 
        for i in range(0, m):
            for j in range(0, n):
                if isWater[i][j] == 1:
                    queue.append((i, j))
                    visited[i][j] = True

        while queue:
            cur = queue.popleft()
            # 周围节点值 赋为 当前节点值 + 1
            for d in dirs:
                x = cur[0] + d[0]
                y = cur[1] + d[1]
                
                if m > x >= 0 and n > y >= 0 and visited[x][y] is False:
                    res[x][y] = res[cur[0]][cur[1]] + 1
                    queue.append((x, y))
                    visited[x][y] = True

        return res