C ++中的墙壁和大门

假设我们有一个mxn 2D网格,并使用这三个可能的值进行了初始化。

  • -1表示墙壁或障碍物。

  • 门为0。

  • INF这是无穷大,表示有一个空房间。

在这里2 ^ 31-1 = 2147483647是INF,因为我们可能假设到门的距离小于2147483647。用到其最近门的距离填充每个空房间。如果不可能到达大门,则应填充INF。

所以,如果输入像

INF-10INF
INFINFINF-1
INF-1INF-1
0-1INFINF

那么输出将是

3-101
221-1
1-12-1
0-134

为了解决这个问题,我们将遵循以下步骤-

  • 定义大小为4 x 2的数组目录:= {{{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, ],]