力扣每日一题0129-1765. 地图中的最高点

149 阅读4分钟

「这是我参与2022首次更文挑战的第7天,活动详情查看:2022首次更文挑战」。

给你一个大小为 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:

image.png

输入:isWater = [[0,1],[0,0]]
输出:[[1,0],[2,1]]
解释:上图展示了给各个格子安排的高度。
蓝色格子是水域格,绿色格子是陆地格。

示例 2:

image.png

输入:isWater = [[0,0,1],[1,0,0],[0,0,0]]
输出:[[1,1,0],[0,1,1],[1,2,2]]
解释:所有安排方案中,最高可行高度为 2 。
任意安排方案中,只要最高高度为 2 且符合上述规则的,都为可行方案。

多源广度优先搜索

题目要求让矩阵中的最高高度最大,我们可以通过最大化每个格子的高度来做到这一点。由于任意相邻的格子高度差至多为 1,这意味着对于每个格子,其高度至多比其相邻格子中的最小高度多 1。

题目要求水域的高度必须为 0,因此水域的高度是已经确定的值,我们可以从水域出发,推导出其余格子的高度:

  • 首先,计算与水域相邻的格子的高度。对于这些格子来说,其相邻格子中的最小高度即为水域的高度 0,因此这些格子的高度为 1。
  • 然后,计算与高度为 1 的格子相邻的、尚未被计算过的格子的高度。对于这些格子来说,其相邻格子中的最小高度为 1,因此这些格子的高度为 2。
  • 以此类推,计算出所有格子的高度。

上述过程中,对于每一轮,我们考虑的是与上一轮格子相邻的、未被计算过的格子 x,其高度必然比上一轮的格子高度多 1。反证之:如果格子 x 的高度小于或等于上一轮的格子高度,那么 x 必然会在更早的轮数被计算出来,矛盾。因此 x 的高度必然比上一轮的格子高度高,同时由于题目要求任意相邻的格子高度差至多为 1,因此 x 的高度必然比上一轮格子的高度多 1。

可以发现,上述过程就是从水域出发,执行广度优先搜索的过程。因此,记录下所有水域的位置,然后执行广度优先搜索,计算出所有陆地格子的高度,即为答案。

/**
 * @param {number[][]} isWater
 * @return {number[][]}
 */
var highestPeak = function(isWater) {
    let m = isWater.length, n = isWater[0].length;
    // 初始化dp
    const dp = Array.from(Array(m), () => Array(n).fill(0));
    // 定一个最大值
    const max = m * n;
    // 先去使用左上到右下的DP
    for(let i = 0; i < m; i++){
        for(let j = 0; j < n; j++){
            if(isWater[i][j] === 1){
                continue;
            }
            // 如果值为0的话,则先去计算它到左上的最小值
            let min = max;
            // 计算上位置的最小值
            if (i > 0) {
                min = Math.min(min, dp[i - 1][j] + 1);
            }
            // 计算左侧位置的最小值
            if (j > 0) {
                min = Math.min(min, dp[i][j - 1] + 1);
            }
            dp[i][j] = min;
        }
    }

    // 从右下到左上的DP
    for (let i = m - 1; i >= 0; i--) {
        for (let j = n - 1; j >= 0; j--) {
            if (i < m - 1) {
                dp[i][j] = Math.min(dp[i][j], dp[i + 1][j] + 1);
            }
            if (j < n - 1) {
                dp[i][j] = Math.min(dp[i][j], dp[i][j + 1] + 1);
            }
        }
    }
    return dp;
};

复杂度分析

  • 时间复杂度:O(mn)O(mn)mmnn 的含义同题目中的定义。
  • 空间复杂度:O(mn)OO(mn)O。最坏情况下整个矩阵都是水域,我们需要将所有水域入队。

image.png