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;// 相当于多源BFSqueue<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--;}}}// 每腐烂一次时间+1if (rotten == true) {ans++;}}if (fresh) {return -1;}return ans;}
};