[LeetCode 1765] 地图中的最高点

题目链接

https://leetcode-cn.com/problems/map-of-highest-peak/

题目描述

给你一个大小为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 个水域格子。

思路

• 把每个单元格看成一个节点，相邻的单元格对应节点认为有一条边权为1的边，于是可以得到一个图；
• 对于每个单元格，它能达到的最大高度，一定不能超过任何一个水域节点到它的最短路长度(否则一定会有相邻的格子高度差大于1)；
• 在图中在引入一个超级节点，假设超级节点和水域节点都有一条边权为1的边，那么每个单元格能达到的最大高度，其实就是到超级节点的最短路长度减1(这也就是单元格节点到所有水域节点的最短路长度的最小值)；
• 可以证明，对于两个相邻的节点，它们到超级节点的最短路长度之差一定小于等于1，因此只要把每个单元格的高度都设置成对应的最大高度(也就是到超级节点的最短路长度减1)，就能够满足题目要求。
• 因为边权都是1，所以可以使用bfs算法计算，实际上算法中不需要显式的定义超级节点，只需要在最初将所有的水域节点加入队列即可。

代码

#include<bits/stdc++.h>
using namespace std;
class Solution {
public:
    int dx[4];
    int dy[4];
    Solution(): dx{0, 0, -1, 1}, dy{-1, 1, 0, 0}{}
    vector<vector<int> > highestPeak(vector<vector<int> >& isWater) {
        int m = isWater.size();
        int n = isWater[0].size();
        vector<vector<int> > height(m, vector<int>(n, -1)); // 数组初始化

        queue<pair<int, int> > Q; // 队列Q
        for(int i=0;i<m;i++)
        {
            for(int j=0;j<n;j++)
            {
                if(isWater[i][j] == 1)  
                {
                    Q.push({i, j});
                    height[i][j] = 0;
                }
            }
        }
        while(!Q.empty())
        {
            pair<int, int> p = Q.front();   Q.pop();
            for(int i=0;i<4;i++) // 遍历四个区域
            {
                int x = p.second + dx[i];
                int y = p.first + dy[i];
                if(y >= 0 && y < m && x >= 0 && x < n)
                {
                    if(height[y][x] == -1)  
                    {
                        height[y][x] = height[p.first][p.second] + 1;
                        Q.push({y, x});
                    }
                }
            }
        }
        return height;
    }
};