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) 的高度。如果有多种解法,请返回 任意一个 。
示例 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 且符合上述规则的,都为可行方案。
广度优先搜索解法:
根据题意能够直接得知,需要返回的结果矩阵中所有水域位置的数据为 0(需要注意输入矩阵 isWater 中的数值 1 表示为水域,而输出结果中水域位置值应为表示高度的 0)。因此首先初始化结果数组 ans,将所有位置的数值初始化为 -1,再遍历 isWater 矩阵将 ans 所有水域位置置为 0,其他位置仍为 -1,表示未处理的陆地区域。
由于任意相邻的格子高度差至多为 1,可以考虑从已经确定的水域位置出发进行广度优先搜索(多元广度优先搜索):
在遍历水域的过程中,依次将水域区域的坐标放入一个队列中,该队列存放所有已经处理过的位置;对于已经处理过的每个位置 (x, y), 理论上有四种后续的前进方向:
正北:(x, y - 1) = (x, y) + (0, -1)
正南:(x, y + 1) = (x, y) + (0, 1)
正东:(x + 1, y) = (x, y) + (1, 0)
正西:(x - 1, y) = (x, y) + (-1, 0)
对于四种可能的后续遍历坐标,需要再判断其是否仍位于矩阵内部,且其之前是否已经处理过(通过 ans 值是否为 -1 进行判断),若符合条件,后续遍历坐标出的值应为当前 (x, y) 坐标位置的值 +1,即任意相邻的格子高度差至多为 1,同时将刚刚处理过的后续遍历坐标再入队用于下一步的广度优先搜索。当所有的位置均处理之后,对于队列中仍剩余的位置,其后续遍历坐标将不再满足条件,不再有坐标入队,队列中的坐标依次出队,直到队列为空结束广度优先搜索。此时 ans 即为需要反馈的存在最大高度的结果。
class Solution {
private:
int directions[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
public:
vector<vector<int>> highestPeak(vector<vector<int>>& isWater) {
int m = isWater.size(), n = isWater[0].size();
vector<vector<int>> ans(m, vector<int>(n, -1));
queue<pair<int, int>> q;
for(int i = 0; i < m; i++){
for(int j = 0; j < n; j++){
if(isWater[i][j] == 1){
ans[i][j] = 0;
q.push({i, j});
}
}
}
while(!q.empty()){
auto p = q.front();
for(auto direction : directions){
int x = p.first + direction[0], y = p.second + direction[1];
if(x >= 0 && x < m && y >= 0 && y < n && ans[x][y] == -1){
ans[x][y] = ans[p.first][p.second] + 1;
q.push({x, y});
}
}
q.pop();
}
return ans;
}
};