腐烂的橘子
Rotting Oranges
解
广度优先搜索(BFS)
class Solution {
public:
int orangesRotting(vector<vector<int>>& grid) {
int count = 0, ret = 0;
int dir_x[4] = { 0,1,0,-1 };
int dir_y[4] = { 1,0,-1,0 };
int n = (int)grid.size();
int m = (int)grid[0].size();
int time[10][10] = { -1 };
bool visited[10][10] = { false };
queue<pair<int, int>>Q;
//memset(time, -1, sizeof(time));
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
if (grid[i][j] == 2) {
Q.push(make_pair(i, j));
time[i][j] = 0;
visited[i][j] = true;
} else if (grid[i][j] == 1) {
count++;
}
}
}
while (!Q.empty()) {
pair<int, int>pos = Q.front();
Q.pop();
for (int i = 0; i < 4; i++) {
int x = pos.first + dir_x[i];
int y = pos.second + dir_y[i];
if (x < 0 || x >= n || y < 0 || y >= m || !grid[x][y] || visited[x][y]) {
continue;
}
time[x][y] = time[pos.first][pos.second] + 1;
Q.push(make_pair(x, y));
visited[x][y] = true;
if (grid[x][y]) {
count--;
ret = time[x][y];
if (!count) {
break;
}
}
}
}
return count ? -1 : ret;
}
};