在给定的 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。
预处理
把初始时的腐烂橘子加入队列,并计算新鲜橘子的个数。
BFS
将预处理阶段找到的腐烂橘子向四个方向开始腐烂,需要记录队列的初始长度,当长度减少到0时算一轮感染。再进行次轮感染,直到队列为空。
class Solution {
public:
int orangesRotting(vector<vector<int>>& grid) {
int fresh=0;
int dir[4][2]={0,1,0,-1,1,0,-1,0};
int m=grid.size();
int n=grid[0].size();
int time=0;
queue<pair<int, int>> que;
for(int i=0;i<m;i++)
{
for(int j=0;j<n;j++)
{
if(grid[i][j]==2)
{
que.push({i,j});
}
else if(grid[i][j]==1)
{
fresh++;
}
}
}
while(!que.empty()) {
int size=que.size();
bool flag=false;
for(int i=0;i<size;i++)
{
pair<int ,int> cur = que.front();
que.pop();
int curx = cur.first;
int cury = cur.second;
for (int i = 0; i < 4; i++) {
int nextx = curx + dir[i][0];
int nexty = cury + dir[i][1];
if (nextx < 0 || nextx >= m || nexty < 0 || nexty >= n||grid[nextx][nexty]==0||grid[nextx][nexty]==2)
{
continue;
}
que.push({nextx,nexty});
grid[nextx][nexty]=2;
fresh--;
flag=true;
}
}
if(flag)
{
time++;
}
}
return fresh ? -1 : time;
}
};