在给定的 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
题目大意:计算所有新鲜橘子被腐烂橘子感染所需的最短时间
分析:
(1)由题得,腐烂橘子会将其四周的新鲜橘子感染,因此可以考虑用广度优先算法计算;
(2)由于一开始可能存在多个腐烂橘子,因此是多源广度优先算法问题,将多个腐烂橘子看做是从一个不存在的腐烂橘子开始扩散得到的,所有将一开始的所有腐烂橘子加入广度优先算法的队列中即可。
class Solution {
public:
int orangesRotting(vector<vector<int>>& grid) {
int m=grid.size(),n=grid[0].size(),step=0,dp[4][2]={{1,0},{-1,0},{0,1},{0,-1}};
queue<pair<int,int>> Q;
for(int i=0;i<m;i++) for(int j=0;j<n;j++) if(grid[i][j]==2) Q.push({i,j});
int size=Q.size();
while(!Q.empty()){
auto [x,y]=Q.front();
Q.pop();
for(int i=0;i<4;i++){
int dx=x+dp[i][0];
int dy=y+dp[i][1];
if(dx>=0&&dx<m&&dy>=0&&dy<n&&grid[dx][dy]==1){
grid[dx][dy]=2;
Q.push({dx,dy});
}
}
if(!(--size)){
//本次腐烂橘子完成扩散后,统计被腐烂的橘子,方便下一次扩散
size=Q.size();
if(size) ++step;//本次扩散有橘子感染,时间+1
}
}
for(int i=0;i<m;i++) for(int j=0;j<n;j++) if(grid[i][j]==1) return -1;
return step;
}
};