其他分享
首页 > 其他分享> > Leetcode 1765. 地图中的最高点(BFS)

Leetcode 1765. 地图中的最高点(BFS)

作者:互联网

给你一个大小为 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

裸BFS,直接以每个水源为原点进行BFS即可。因为要求高度尽可能大,那么每次访问到的每个位置周围的四个位置高度直接设置为当前位置+1(如果还没被访问过的话)。容易证明这样不会出现断崖的情况(类比一下等高线)。一开始代码中vis数组没判断全导致wa了几发==

class Solution {
public: 
    #define fi first
    #define se second
    int a[1005][1005], n, m;
    queue<pair<pair<int, int>, int> > q;
    int dir[4][2] = {{0, 1}, {0, -1}, {1, 0}, {-1, 0}};
    bool vis[1005][1005];
    void bfs() {
        while(q.size()) {
            pair<pair<int, int>, int> now = q.front();
            q.pop();
            if(vis[now.fi.fi][now.fi.se]) continue;
            a[now.fi.fi][now.fi.se] = now.se;
            vis[now.fi.fi][now.fi.se] = 1;
            for(int i = 0; i < 4; i++) {
                int nx = now.fi.fi + dir[i][0], ny = now.fi.se + dir[i][1];
                if(nx > n || nx < 1 || ny > m || ny < 1 || vis[nx][ny]) continue;
                q.push(make_pair(make_pair(nx, ny), now.se + 1));
            }
        }
    }
    vector<vector<int>> highestPeak(vector<vector<int>>& isWater) {
        memset(vis, 0, sizeof(vis));
        n = isWater.size(), m = isWater[0].size();
        for(int i = 0; i < n; i++) {
            for(int j = 0; j < m; j++) {
                a[i + 1][j + 1] = -isWater[i][j];
                if(a[i + 1][j + 1] == -1) {
                    q.push(make_pair(make_pair(i + 1, j + 1), 0));
                } else a[i + 1][j + 1] = 0;
            }
        }
        bfs();
        vector<vector<int> > ans(n);
        for(int i = 0; i < n; i++) {
            for(int j = 0; j < m; j++) {
                if(a[i + 1][j + 1] != -1) ans[i].push_back(a[i + 1][j + 1]);
                else ans[i].push_back(0);
            }
        }
        return ans;
    }
};

标签:格子,int,isWater,BFS,vis,fi,now,Leetcode,1765
来源: https://www.cnblogs.com/lipoicyclic/p/15854838.html