LeetCode 542. 01 Matrix

输入：只包含0，1的矩阵

输出：元素1到达最近0的距离

算法思想：广度优先搜索。

元素为0为可达区域，元素为1为不可达区域，我们的目标是为了从可达区域不断地扩展至不可达区域，在扩展的过程中，也就计算出了这些不可达区域到达最近可达区域的距离。

每个可达元素都记录了到当前位置的距离，因此在后续的遍历中，如果是经由当前节点到达的下一节点，这个距离会被累加。

 #include <iostream>
 #include <vector>
 #include <queue>
 using namespace std;

 class Solution {
 public:
     vector<vector<int>> updateMatrix(vector<vector<int>>& matrix) {
         int nRow = matrix.size();
         int nCol = matrix[].size();
         vector<vector<int>> answer(nRow, vector<int>(nCol));
         queue<pair<int, int>> reachable;
         for (int i = ; i < nRow; i++)
         {
             for (int j = ; j < nCol; j++)
             {
                 if (matrix[i][j] == ) // reachable
                 {
                     reachable.push(make_pair(i, j));
                     answer[i][j] = ;
                 }
                 else
                     answer[i][j] = INT_MAX;
             }
         }

         vector<pair<int, int>> dir = vector<pair<int, int>>({make_pair(-,),make_pair(,),make_pair(,-),make_pair(,)});

         while (!reachable.empty())
         {
             pair<int, int> cur = reachable.front();
             for (int i = ; i < ; i++)
             {
                 int x = dir[i].first;
                 int y = dir[i].second;
                 int cx = cur.first;
                 int cy = cur.second;
                 if (cx + x <  || cx + x > nRow -  || cy + y <  || cy + y > nCol - ) // boundary test
                     continue;
                 if (matrix[cx+x][cy+y] == ) // not visited
                 {
                     matrix[cx + x][cy + y] = ; // label visited
                     answer[cx + x][cy + y] = answer[cx][cy] + ;
                     reachable.push(make_pair(cx + x, cy + y));
                 }
             }
             reachable.pop();
         }
         return answer;
     }
 };