[LeetCode] Walls and Gates

286. Walls and Gates

You are given an m x n grid rooms initialized with these three possible values.

  • -1 A wall or an obstacle.
  • 0 A gate.
  • INF Infinity means an empty room. We use the value 231 - 1 = 2147483647 to represent INF as you may assume that the distance to a gate is less than 2147483647.

Fill each empty room with the distance to its nearest gate. If it is impossible to reach a gate, it should be filled with INF.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
class Solution {
public:
void wallsAndGates(vector<vector<int>>& rooms) {
int n = rooms.size(), m = rooms[0].size(), INF = 2147483647;
int dx[4] = {0,1,0,-1}, dy[4] = {-1,0,1,0};
queue<pair<int,int>> q;
for(int i = 0; i < n; i++) {
for(int j = 0; j < m; j++) {
if(!rooms[i][j])
q.push({i,j});
}
}
while(!q.empty()) {
auto [y, x] = q.front(); q.pop();
for(int i = 0; i < 4; i++) {
int ny = y + dy[i], nx = x + dx[i];
if(0 <= nx and nx < m and 0 <= ny and ny < n and rooms[ny][nx] == INF) {
rooms[ny][nx] = rooms[y][x] + 1;
q.push({ny,nx});
}
}
}
}
};
Author: Song Hayoung
Link: https://songhayoung.github.io/2022/02/20/PS/LeetCode/walls-and-gates/
Copyright Notice: All articles in this blog are licensed under CC BY-NC-SA 4.0 unless stating additionally.