【算法分析与设计】【第十二周】542. 01 Matrix

题目来源：542：https://leetcode.com/problems/01-matrix/description/

BFS基础训练。

542. 01 Matrix

题目大意

在0/1矩阵中，求出所有元素的距最近0的距离。

例1
Input:
0 0 0
0 1 0
0 0 0

Output:
0 0 0
0 1 0
0 0 0

例2
Input:
0 0 0
0 1 0
1 1 1

Output:
0 0 0
0 1 0
1 2 1

---

思路

其实本质是求BFS的层数。
做BFS的时候记录层数就可以了。
这里和常规思维有点不同的是，入队的是0。

---

解题代码

class Solution {public:vector<vector<int>> updateMatrix(vector<vector<int>>& matrix) {}};// 点数据结构 struct Point {  int x;  int y;  Point(int _x, int _y) : x(_x), y(_y) {}};// 代表四个方向 int Posx[4] = {-1, 0, 1, 0};int Posy[4] = {0, -1, 0, 1};class Solution {public:  bool isValid(int x, int y, vector<vector<int>>& A) {  return x >= 0 && x < A.size() &&  y >= 0 && y < A[x].size();  }vector<vector<int>> updateMatrix(vector<vector<int>>& A) {   queue<Point> q1, q2;        // 两个保存同节点的队列  int step = 1;               // 表示走到0所需的步数，遍历一个队列赋一次值   // 将1更新成-1，即未赋值，将0入队  for (int i = 0; i < A.size(); i++)  for (int j = 0; j < A[i].size(); j++)  if (A[i][j] == 0)  q1.push(Point(i, j));.  else  A[i][j] = -1;  while (!q1.empty() || !q2.empty()) {    while (!q1.empty()) {      Point p = q1.front();      q1.pop();      // 添加四个方向的节点       for (int i = 0; i < 4; i++) {        int newx = p.x + Posx[i];        int newy = p.y + Posy[i];        if (isValid(newx, newy, A) && A[newx][newy] == -1) {          A[newx][newy] = step;          q2.push(Point(newx, newy));        }      }    }    step++;    while (!q2.empty()) {      Point p = q2.front();      q2.pop();      // 添加四个方向的节点       for (int i = 0; i < 4; i++) {        int newx = p.x + Posx[i];        int newy = p.y + Posy[i];        if (isValid(newx, newy, A) && A[newx][newy] == -1) {          A[newx][newy] = step;          q1.push(Point(newx, newy));        }     }   }   step++;  }  return A;  }};

---

时间复杂度

O(rows*cols)

---