1765. 地图中的最高点

Alex_Shen
2022-03-29 / 0 评论 / 0 点赞 / 122 阅读 / 1,226 字 / 正在检测是否收录...
温馨提示:
本文最后更新于 2022-03-31,若内容或图片失效,请留言反馈。部分素材来自网络,若不小心影响到您的利益,请联系我们删除。

给你一个大小为 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:

img

输入:isWater = [[0,1],[0,0]]
输出:[[1,0],[2,1]]
解释:上图展示了给各个格子安排的高度。
蓝色格子是水域格,绿色格子是陆地格。

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

思路:

多源bfs:从每一个水源开始向外扩展,如果没有被填入过,就是中心高度+1

代码:

class Solution {
public:
    vector<vector<int>> highestPeak(vector<vector<int>>& isWater) {
        int m=isWater.size(),n=isWater[0].size();
        queue<vector<int>> q;
        vector<vector<int>> mp(m,vector<int>(n,-1));
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                if(isWater[i][j]==1){
                    mp[i][j]=0;
                    q.push(vector<int>{i,j});
                }
            }
        }
        int direction[4][2] = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};
        while(!q.empty()){
            auto &p = q.front();
            for(int i=0;i<4;i++){
                int x=p[0]+direction[i][0],y=p[1]+direction[i][1];
                if(x>=0&&x<m&&y>=0&&y<n&&mp[x][y]==-1){
                    mp[x][y]=mp[p[0]][p[1]]+1;
                    q.push(vector<int>{x,y});
                }
            }
            q.pop();
        }
        return mp;
    }
};
0

评论区