给你一个大小为 m x n 的整数矩阵 isWater ,它代表了一个由 陆地 和 水域 单元格组成的地图。
如果 isWater[i][j] == 0
,格子 (i, j) 是一个 陆地 格子。
如果 isWater[i][j] == 1
,格子 (i, j) 是一个 水域 格子。
你需要按照如下规则给每个单元格安排高度:
找到一种安排高度的方案,使得矩阵中的最高高度值 最大 。
请你返回一个大小为 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 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
public:
vector<vector<int>> highestPeak(vector<vector<int>>& isWater) {
int m = isWater.size(), n = isWater[0].size();
queue<pair<int,int>> q;
vector<vector<bool>> vis(m, vector<bool>(n, false));
for(int i = 0; i < m; i++) {
for(int j = 0; j < n; j++)
if(isWater[i][j]) // 是水
{
q.push({i,j});
vis[i][j] = true;
}
}
vector<vector<int>> dir = {{1,0},{0,1},{-1, 0},{0,-1}};
int h = 0;
while(!q.empty())
{
int size = q.size();
while(size--)
{
pair<int,int> t = q.front();
q.pop();
isWater[t.first][t.second] = h;
for(int k = 0; k < 4; ++k)
{
int x = t.first+dir[k][0];
int y = t.second+dir[k][1];
if(x>=0 && x < m && y>=0 && y < n && !vis[x][y])
{
q.push({x,y});
vis[x][y] = true;
}
}
}
h++;
}
return isWater;
}
};
464 ms 106.4 MB C++