994. 腐烂的橘子(面试题打卡/前缀和/简单)
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/rotting-oranges/
题干:
在给定的 m x n
网格 grid
中,每个单元格可以有以下三个值之一:
- 值
0
代表空单元格; - 值
1
代表新鲜橘子; - 值
2
代表腐烂的橘子。
每分钟,腐烂的橘子 周围 4 个方向上相邻 的新鲜橘子都会腐烂。
返回 直到单元格中没有新鲜橘子为止所必须经过的最小分钟数。如果不可能,返回 -1
。
提示:
m == grid.length
n == grid[i].length
1 <= m, n <= 10
grid[i][j]
仅为0
、1
或2
示例:
输入:grid = [[2,1,1],[1,1,0],[0,1,1]]
输出:4
输入:grid = [[2,1,1],[0,1,1],[1,0,1]]
输出:-1
解释:左下角的橘子(第 2 行, 第 0 列)永远不会腐烂,因为腐烂只会发生在 4 个正向上。
输入:grid = [[0,2]]
输出:0
解释:因为 0 分钟时已经没有新鲜橘子了,所以答案就是 0 。
题解:
广度优先遍历:
使用一个队列来保存腐烂橘子的坐标。首先,遍历整个网格,将腐烂橘子的坐标加入队列,并统计新鲜橘子的数量。然后,开始进行腐烂,每一轮从队列中取出腐烂橘子的坐标,遍历其四个方向,将新鲜橘子标记为腐烂,并将其坐标加入队列。每一轮腐烂后,分钟数加一。最后,如果还有新鲜橘子剩余,则返回 -1,否则返回腐烂的分钟数。
class Solution {
public static int orangesRotting(int[][] grid) {
int n = grid.length, m = grid[0].length;
// 记录新鲜橘子数
int freshOranges = 0;
// 保存腐烂橘子的坐标
Queue<int[]> queue = new LinkedList<>();
// 遍历网格,将腐烂橘子坐标入队,并统计新鲜橘子数量
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
if (grid[i][j] == 2) {
queue.offer(new int[]{i, j});
} else if (grid[i][j] == 1) {
freshOranges++;
}
}
}
// 如果没有新鲜橘子,则不需要进行腐烂
if (freshOranges == 0) {
return 0;
}
int minutes = -1;
int[] dx = {-1, 0, 1, 0}, dy = {0, 1, 0, -1};
// 开始腐烂
while (!queue.isEmpty()) {
int size = queue.size();
for (int i = 0; i < size; i++) {
int[] orange = queue.poll();
// 上下左右
for (int j = 0; j < 4; j++) {
int x = orange[0] + dx[j], y = orange[1] + dy[j];
// 如果新的坐标越界或不是新鲜橘子,则越过
if (x < 0 || x >= n || y < 0 || y >= m || grid[x][y] != 1) {
continue;
}
// 将新鲜橘子标记为腐烂,并入队
grid[x][y] = 2;
queue.offer(new int[]{x, y});
freshOranges--;
}
}
// 每一轮腐烂后,分钟数加一
minutes++;
}
// 如果还有新鲜橘子剩余,则返回 -1
if (freshOranges > 0) {
return -1;
}
// 返回腐烂的分钟数
return minutes;
}
}