A.门牌制作
题目描述
给你一个大小为 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:
输入: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 个水域格子。
解题思路
动态规划
- 我们可以 想到当isWater[i][j] != 1 时,即不是水域时,该片陆地的高度dp[i][j] = min(dp[i-1][j]+1, dp[i+1][j]+1, dp[i][j-1]+1, dp[i][j+1]+1)
- 当isWater[i][j] == 1 时,即是水域时,dp[i][j] = 0
- 要求dp[i][j] = min(dp[i-1][j]+1, dp[i+1][j]+1, dp[i][j-1]+1, dp[i][j+1]+1),只需要从左上到右下,求右,下两个区域的最小值,再从左下到右上求右,上两个区域的最小值。
python代码
def highestPeak(self, isWater: List[List[int]]) -> List[List[int]]:
n,m = len(isWater),len(isWater[0])
Max = n*m
dp = [[0]*m for _ in range(n)]
for i in range(n):
for j in range(m):
if(isWater[i][j] == 1):
continue
Min = Max
if i > 0:
Min = min(Min,dp[i-1][j] + 1)
if j > 0:
Min = min(Min,dp[i][j-1] + 1)
dp[i][j] = Min
print(dp)
for i in range(n-1,-1,-1):
for j in range(m-1,-1,-1):
if i < n - 1:
dp[i][j] = min(dp[i][j],dp[i+1][j]+1)
if j < m - 1:
dp[i][j] = min(dp[i][j],dp[i][j+1]+1)
return dp