携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第30天，点击查看活动详情 >>

地图中的最高点

给你一个大小为 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 个水域格子。

解题思路:

1. 从水域开始，广度优先搜索，后一层的高度为前一层的高度+1；
2. 注意： 用队列APIshift()会超时，解决方法是每一层单独开一个新栈，用pop()和push()操作代替shift()。

我的答案:

/**
 * @param {number[][]} isWater
 * @return {number[][]}
 */
var highestPeak = function(isWater) {
    // 0 -> land   1 -> water
    let h = isWater.length, w = isWater[0].length;
    let q = [];
    for (let r=0; r<h; r++) {
        for (let c=0; c<w; c++) {
            if(isWater[r][c] === 1) {
                isWater[r][c] = 0;
                q.push([r,c]);
            } // water
            else isWater[r][c] = -1; // land
        }
    }
    let pos = [[-1,0],[1,0],[0,-1],[0,1]];
    while (q.length) {
        let next = [];
        while (q.length) {
            let [cr,cc] = q.pop();
            for (let [x,y] of pos) {
                if (x+cr>=0 && x+cr<h && cc+y >=0 && cc+y <w && isWater[x+cr][y+cc] === -1) {
                    isWater[cr+x][cc+y] = isWater[cr][cc] + 1;
                    next.push([cr+x,cc+y]);
                }
            }
        }
        q = next;
    }
    return isWater;
};

最后

如果有更好的解法或者思路, 欢迎在评论区和我交流~ ღ( ´･ᴗ･` )