题目描述:
你被给定一个 m × n 的二维网格,网格中有以下三种可能的初始化值:
-1 表示墙或是障碍物
0 表示一扇门
INF 无限表示一个空的房间。然后,我们用 231 - 1 = 2147483647 代表 INF。你可以认为通往门的距离总是小于 2147483647 的。
你要给每个空房间位上填上该房间到 最近 门的距离,如果无法到达门,则填 INF 即可。
示例:
给定二维网格:
INF -1 0 INF
INF INF INF -1
INF -1 INF -1
0 -1 INF INF
运行完你的函数后,该网格应该变成:
3 -1 0 1
2 2 1 -1
1 -1 2 -1
0 -1 3 4
方法1:
主要思路:
(1)使用广度优先搜索,从门的 0 开始,每次在其相邻的位置大于当前位置的元素值加1时,进行更新该位置的值;
class Solution {
public:
void wallsAndGates(vector<vector<int>>& rooms) {
//处理特殊的情形
if(rooms.empty()||rooms[0].empty()){
return;
}
int rows=rooms.size();
int cols=rooms[0].size();
//初始化广度优先的队列
queue<pair<int,int>> q;
for(int i=0;i<rows;++i){
for(int j=0;j<cols;++j){
if(rooms[i][j]==0){
q.push(pair<int,int>(i,j));
}
}
}
//可能移动的步骤
int step_x[4]={0,1,0,-1};
int step_y[4]={1,0,-1,0};
while(!q.empty()){
//当前位置
int x=q.front().first;
int y=q.front().second;
q.pop();
//可能的四个相邻的位置
for(int i=0;i<4;++i){
int new_x=x+step_x[i];
int new_y=y+step_y[i];
//相邻位置满足要求的情形下,进行更新
if(new_x>=0&&new_x<rows&&new_y>=0&&new_y<cols&&rooms[new_x][new_y]>rooms[x][y]+1){
rooms[new_x][new_y]=rooms[x][y]+1;
q.push(pair<int,int>(new_x,new_y));
}
}
}
return;
}
};