994 腐烂的橘子
在给定的 m x n 网格 grid 中,每个单元格可以有以下三个值之一:
值 0 代表空单元格;
值 1 代表新鲜橘子;
值 2 代表腐烂的橘子。
每分钟,腐烂的橘子 周围 4 个方向上相邻 的新鲜橘子都会腐烂。
返回 直到单元格中没有新鲜橘子为止所必须经过的最小分钟数。如果不可能,返回 -1 。
示例 1:
输入:grid = [[2,1,1],[1,1,0],[0,1,1]]
输出:4
示例 2:
输入:grid = [[2,1,1],[0,1,1],[1,0,1]]
输出:-1
解释:左下角的橘子(第 2 行, 第 0 列)永远不会腐烂,因为腐烂只会发生在 4 个方向上。
示例 3:
输入:grid = [[0,2]]
输出:0
解释:因为 0 分钟时已经没有新鲜橘子了,所以答案就是 0 。
提示:
m == grid.length
n == grid[i].length
1 <= m, n <= 10
grid[i][j] 仅为 0、1 或 2
思路(来自精选题解-腐烂的橘子)
本题采用多源BFS法,因为最初腐烂的橘子可能不止1个,所以需要将传统的BFS修改一下。刚开始还要记录新鲜橘子数fresh,每次需要更新,最后看fresh是否不为0,不为0表示未全部腐烂,返回-1.
传统BFS(胡凡《算法笔记》)
这里需要最初将所有腐烂的橘子放入队列中,然后每次遍历该层的所有腐烂橘子,而不是每次取一个。
每腐烂一次时间+1。
代码
class Solution {
public:
struct node {
int x, y;
} Node;
int X[4] = {0, 0, 1, -1};
int Y[4] = {1, -1, 0, 0};
int orangesRotting(vector<vector<int>>& grid) {
int m = grid.size(); // 行数
int n = grid[0].size(); // 列数
int ans = 0, fresh = 0;
// 相当于多源BFS
queue<node> q;
for (int x = 0; x < m; x++) {
for (int y = 0; y < n; y++) {
// 记录新鲜橘子数
if (grid[x][y] == 1) {
fresh++;
}
// 把腐烂橘子都放入队列中
if (grid[x][y] == 2) {
Node.x = x;
Node.y = y;
q.push(Node);
}
}
}
// 队列q中存放的都是腐烂的橘子
while (!q.empty()) {
int num = q.size();
bool rotten = false;
for (int i = 0; i < num; i++) {
node top = q.front();
q.pop();
for (int i = 0; i < 4; i++) {
int newX = top.x + X[i];
int newY = top.y + Y[i];
// 不出界以及不腐烂
if (newX >= 0 && newX < m && newY >= 0 && newY < n &&
grid[newX][newY] == 1) {
grid[newX][newY] = 2;
Node.x = newX;
Node.y = newY;
q.push(Node);
rotten = true;
fresh--;
}
}
}
// 每腐烂一次时间+1
if (rotten == true) {
ans++;
}
}
if (fresh) {
return -1;
}
return ans;
}
};