Leetcode 1765 地图中的最高点

题目

给你一个大小为 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) 的高度。如果有多种解法，请返回 任意一个 。

解题思路

  多源BFS。题目中要求任意相邻节点高度差不能超过1，又想使高度尽可能的高，那么对于每个水域向四周蔓延，每次遇到未到达的节点都使高度+1。整个过程可以将所有水域作为BFS的起点，搜索的过程就是更新高度的过程。

代码

class Node {
    int x, y, step;

    public Node(int x, int y, int step) {
        this.x = x;
        this.y = y;
        this.step = step;
    }
}

class Solution {
    public static final int[][] turn = new int[][]{{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
    private boolean[][] flag;
    private int row;
    private int col;

    public int[][] highestPeak(int[][] isWater) {
        row = isWater.length;
        col = isWater[0].length;
        flag = new boolean[row][col];

        Queue<Node> queue = new ArrayDeque<>();
        for (int i = 0; i < row; i++) {
            for (int j = 0; j < col; j++) {
                if (isWater[i][j] == 1) {
                    queue.offer(new Node(i, j, 0));
                    isWater[i][j] = 0;
                    flag[i][j] = true;
                }
            }
        }
        while (!queue.isEmpty()) {
            Node now = queue.poll();
            for (int i = 0; i < 4; i++) {
                Node next = new Node(now.x + turn[i][0], now.y + turn[i][1], now.step + 1);
                if (judge(next)) continue;
                isWater[next.x][next.y] = next.step;
                flag[next.x][next.y] = true;
                queue.offer(next);
            }
        }
        return isWater;
    }

    private boolean judge(Node next) {
        if (next.x < 0 || next.x >= row || next.y < 0 || next.y >= col) return true;
        return flag[next.x][next.y];
    }
}