LeetCode——1765. 地图中的最高点(Map of Highest Peak)[中等]——分析及代码（Java）

一、题目

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/map-of-highest-peak
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、分析及代码

1. 按高度逐层添加

（1）思路

从高度为 0 的水域开始，遍历当前 h 高度格子周围的未开发格子，将其高度记为 h + 1，逐层添加直至所有格子处理完成。

（2）代码

class Solution {
    public int[][] highestPeak(int[][] isWater) {
        int m = isWater.length, n = isWater[0].length;
        int[][] ans = new int[m][n];
        for (int i = 0; i < m; i++)
            for (int j = 0; j < n; j++)
                ans[i][j] = -1;//-1表示格子未处理
        
        Queue<Integer> q = new LinkedList<Integer>();//点坐标 x * 1000 + y
        int num = 0;
        for (int i = 0; i < m; i++)
            for (int j = 0; j < n; j++)
                if (isWater[i][j] == 1) {//处理0高度格子
                    ans[i][j] = 0;
                    q.offer(i * 1000 + j);
                    num++;
                }
        
        int h = 1;
        while (!q.isEmpty()) {
            int num0 = num;//当前高度待处理格子数
            num = 0;
            for (int i = 0; i < num0; i++) {
                int p = q.poll();
                int x = p / 1000, y = p % 1000;//还原坐标
                if (x > 0 && ans[x - 1][y] == -1) {//左格子
                    ans[x - 1][y] = h;
                    q.offer((x - 1) * 1000 + y);
                    num++;
                }
                if (x < m - 1 && ans[x + 1][y] == -1) {//右格子
                    ans[x + 1][y] = h;
                    q.offer((x + 1) * 1000 + y);
                    num++;
                }
                if (y > 0 && ans[x][y - 1] == -1) {//上格子
                    ans[x][y - 1] = h;
                    q.offer(x * 1000 + y - 1);
                    num++;
                }
                if (y < n - 1 && ans[x][y + 1] == -1) {//下格子
                    ans[x][y + 1] = h;
                    q.offer(x * 1000 + y + 1);
                    num++;
                }
            }
            h++;//准备处理下一高度格子
        }
        
        return ans;
    }
}

（3）结果

执行用时 ：60 ms，在所有 Java 提交中击败了 84.37% 的用户；
内存消耗 ：155.5 MB，在所有 Java 提交中击败了 20.32% 的用户。

三、其他

暂无。