C ++中的墙壁和大门
假设我们有一个mxn2D网格,并使用这三个可能的值进行了初始化。
-1表示墙壁或障碍物。
门为0。
INF这是无穷大,表示有一个空房间。
在这里2^31-1=2147483647是INF,因为我们可能假设到门的距离小于2147483647。用到其最近门的距离填充每个空房间。如果不可能到达大门,则应填充INF。
所以,如果输入像
那么输出将是
为了解决这个问题,我们将遵循以下步骤-
定义大小为4x2的数组目录:={{{1,0},{-1,0},{0,1},{0,-1}}
n:=房间大小
m:=(如果n为非零,则为列数,否则为0)
定义一对队列q
对于初始化i:=0,当i<n时,更新(将i增加1),执行-
如果rooms[i,j]等于0,则-
将{i,j}插入q
对于初始化j:=0,当j<m时,更新(将j增加1),执行-
对于初始化lvl:=1,当非q为空时,更新(将lvl增加1),执行-
定义一对curr:=q的第一个元素
从q删除元素
x:=curr.first
y:=curr.second
对于初始化i:=0,当i<4时,更新(将i增加1),请执行-
忽略以下部分,跳至下一个迭代
nx:=x+dir[i,0]
ny:=y+dir[i,1]
如果nx<0或ny<0或nx>=n或ny>=m或rooms[nx,ny]<lvl,则-
rooms[nx,ny]:=lvl
将{nx,ny}插入q
sz:=q的大小
当sz为非零值时,请在每次迭代中将sz减1,然后执行-
例
让我们看下面的实现以更好地理解-
#include <bits/stdc++.h> using namespace std; void print_vector(vector<vector<auto< > v){ cout << "["; for(int i = 0; i<v.size(); i++){ cout << "["; for(int j = 0; j <v[i].size(); j++){ cout << v[i][j] << ", "; } cout << "],"; } cout << "]"<<endl; } int dir[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}}; class Solution { public: void wallsAndGates(vector<vector<int<>& rooms) { int n = rooms.size(); int m = n ? rooms[0].size() : 0; queue<pair<int, int> > q; for (int i = 0; i < n; i++) { for (int j = 0; j < m; j++) { if (rooms[i][j] == 0) q.push({ i, j }); } } for (int lvl = 1; !q.empty(); lvl++) { int sz = q.size(); while (sz--) { pair<int, int> curr = q.front(); q.pop(); int x = curr.first; int y = curr.second; for (int i = 0; i < 4; i++) { int nx = x + dir[i][0]; int ny = y + dir[i][1]; if (nx < 0 || ny < 0 || nx >= n || ny >= m || rooms[nx][ny] < lvl) continue; rooms[nx][ny] = lvl; q.push({ nx, ny }); } } } } }; main(){ vector<vector<int<> v = {{2147483647,-1,0,2147483647}, {2147483647,2147483647,2147483647,-1}, {2147483647,-1,2147483647,-1}, {0,-1,2147483647,2147483647}}; Solution ob; ob.wallsAndGates(v); print_vector(v); }
输入项
{{2147483647,-1,0,2147483647},{2147483647,2147483647,2147483647,-1}, {2147483647,-1,2147483647,-1},{0,-1,2147483647,2147483647}}
输出结果
[[3, -1, 0, 1, ],[2, 2, 1, -1, ],[1, -1, 2, -1, ],[0, -1, 3, 4, ],]