LeetCode 5671. 地图中的最高点(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) 的高度。
如果有多种解法,请返回 任意一个 。

数据范围:
m == isWater.length
n == isWater[i].length
1 <= m, n <= 1000
isWater[i][j] 要么是 0 ,要么是 1 。
至少有 1 个水域格子。

解法:

由于水的位置必须为0,那么与水相邻的必须为1...
将所有水的位置丢入队列,bfs一下即可.

code:

#define PI pair<int,int>
int dx[]={0,0,-1,1};
int dy[]={1,-1,0,0};
class Solution {
public:
    vector<vector<int>> highestPeak(vector<vector<int>>& a) {
        int n=a.size(),m=a[0].size();
        vector<vector<int> >mark(n,vector<int>(m,0));
        queue<PI>q;
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                if(a[i][j]==1){
                    q.push({i,j});
                    mark[i][j]=1;
                }
            }
        }
        while(q.size()){
            PI t=q.front();q.pop();
            int x=t.first,y=t.second;
            for(int k=0;k<4;k++){
                int xx=x+dx[k];
                int yy=y+dy[k];
                if(xx<0||xx>=n||yy<0||yy>=m)continue;
                if(mark[xx][yy])continue;
                mark[xx][yy]=mark[x][y]+1;
                q.push({xx,yy});
            }
        }
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                mark[i][j]--;
            }
        }
        return mark;
    }
};

上一篇:JAVA对象分析之偏向锁、轻量级锁、重量级锁升级过程


下一篇:百度飞桨领航团零基础Python速成营 课程总结5