关于广度优先搜索的问题。
思路:
因为这道题是求值为1的点到值为0的区域的最短距离,因为是在图上求最短距离,所以可以用广度优先搜索方法求解,该方法在求最小生成树和迪杰斯特拉最短路径算法的思想中都有体现。本质上是有一个已知最优路线(最优一般指距离最短)区域,这个区域被放进queue中,这道题的这个区域就是0区域,所以一开始要将0区域都找出来,并把它们push进queue里,然后开始BFS,每遇到一个未访问过的点(ret[i][j]值为-1)和合法的点(不超过map)的区域,就将它加一并push进queue中,这样,图map总会被遍历完,而且总是最小值。经验:
四个方向遍历可以声明一个大小为4的pair数组来使用。
熟练使用typedef
C++代码:
class Solution {
public:
vector<vector<int>> updateMatrix(vector<vector<int>>& matrix) {
int rows = matrix.size();
int clos = matrix[0].size();
vector<vector<int>> ret = matrix;
typedef pair<int, int> point;
queue<point> q;
for (int i=0; i<rows; i++) {
for (int j=0; j<clos; j++) {
if (matrix[i][j] == 0){
ret[i][j] = 0;
q.push(point(i,j));
}
else
ret[i][j] = -1;
}
}
point d[4] = {point(-1,0),point(1,0),point(0,-1),point(0,1)};
while (q.size()) {
point temp = q.front();
q.pop();
int x = temp.first;
int y = temp.second;
for (int i=0; i<4; i++) {
int xx = x + d[i].first;
int yy = y + d[i].second;
if (xx>=0 && xx<rows && yy>=0 && yy<clos && ret[xx][yy]==-1) {
ret[xx][yy] = ret[x][y] + 1;
q.push(point(xx,yy));
}
}
}
return ret;
}
};