LeetCode 1765. 地图中的最高点（BFS）

Michael阿明 2021-02-21 16:31:55

分类专栏： LeetCode

本文链接：https://blog.csdn.net/qq_21201267/article/details/113919848

版权

文章目录

- 1. 题目
- 2. 解题

1. 题目

给你一个大小为 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 且符合上述规则的，都为可行方案。
 
提示：
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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

广度优先搜索即可，先把水域加入队列

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;
    }
};