LeetCode-1765 地图中的最高点

来源：力扣（LeetCode）
链接：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 个水域格子。

 

解题思路

一道标注的BFS题，已知水面的高度必然为0，而相邻地块的高度差只能为1，所以在已知水域位置的情况下，地表高度其实是固定的，从水域出发，像附近的地块依次进行高度差为1的扩展，最后扩展出的地形就是所求的地形。

代码展示

class Solution {
public:
    vector<vector<int>> highestPeak(vector<vector<int>>& isWater) {
        vector<vector<int>> vviRet;
        int iM = isWater.size();
        int iN = isWater[0].size();
        vector<int> viTemp;
        viTemp.resize(iN, -1);
        vviRet.resize(iM, viTemp);
        queue<pair<int, int>> queuepairiiVisiting;
        for(int i = 0; i < iM; i ++)
        {
            for(int j = 0; j < iN; j++)
            {
                if(isWater[i][j])
                {
                    vviRet[i][j] = 0;
                    queuepairiiVisiting.push({i, j});
                }
            }
        }
        while(!queuepairiiVisiting.empty())
        {
            int i = queuepairiiVisiting.front().first;
            int j = queuepairiiVisiting.front().second;
            queuepairiiVisiting.pop();
            int iHeight = vviRet[i][j];
            if(i - 1 >= 0 && vviRet[i - 1][j] == -1)
            {
                vviRet[i - 1][j] = iHeight + 1;
                queuepairiiVisiting.push({i - 1, j});
            }
            if(i + 1 < iM && vviRet[i + 1][j] == -1)
            {
                vviRet[i + 1][j] = iHeight + 1;
                queuepairiiVisiting.push({i + 1, j});
            }
            if(j - 1 >= 0 && vviRet[i][j - 1] == -1)
            {
                vviRet[i][j - 1] = iHeight + 1;
                queuepairiiVisiting.push({i, j - 1});
            }
            if(j + 1 < iN && vviRet[i][j + 1] == -1)
            {
                vviRet[i][j + 1] = iHeight + 1;
                queuepairiiVisiting.push({i, j + 1});
            }
        }
        return vviRet;
    }
};

 

运行结果