给你一个大小为 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) 的高度。如果有多种解法,请返回 任意一个 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/map-of-highest-peak
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
bfs
class Solution {
public int[][] highestPeak(int[][] isWater) {
if(isWater == null || isWater.length == 0)
return new int[][]{};
Queue<int[]> queue = new LinkedList<>();
int m = isWater.length;
int n = isWater[0].length;
int[][] res = new int[m][n];
for(int i = 0;i < m; i++){
Arrays.fill(res[i], -1);
}
for(int i = 0;i < m; i++){
for(int j = 0; j < n; j++){
if(isWater[i][j] == 1){
queue.offer(new int[]{i,j});
res[i][j] = 0;
}
}
}
int[][] dirs = new int[][]{{1,0},{-1,0},{0,1},{0,-1}};
int height = 0;
while(!queue.isEmpty()){
int size = queue.size();
for(int i = 0; i < size; i++){
int[] temp = queue.poll();
int x = temp[0];
int y = temp[1];
for(int dir[]:dirs){
int dx = x + dir[0];
int dy = y + dir[1];
if(dx < 0 || dx >= m || dy < 0 || dy >= n)
continue;
if(res[dx][dy] != -1)
continue;
res[dx][dy] = height + 1;
queue.offer(new int[]{dx, dy});
}
}
height++;
}
return res;
}
}