题目传送门:994. 腐烂的橘子 - 力扣(LeetCode)
题解
题目大意为给定一张m×n的网格,每个格中可能是新鲜橘子,腐烂橘子或者空,腐烂橘子每分钟会使其上下左右的四个位置的新鲜橘子腐烂,求多少分钟后格子中没有新鲜橘子。
根据腐烂橘子扩散的特性与广度优先搜索的方法相同,本文将使用广搜完成。
首先对网格进行预处理,分别将好橘子和烂橘子的坐标记录到两个队列que中,此处用数组模拟队列,在记录是还使用到了一个小技巧,如果同时记录行列信息需要两个变量,根据数据范围行列都小于等于10,所以我们将行列拼成一个四位数,例如3行5列记录为0305=3*100+5,10行1列记为1001=10*100+1。
接下来我们开始模拟腐烂扩散的过程,其中timee[i]记录队列中第i个橘子是第几分钟腐烂的。采用广搜的方法,每次从烂橘子队列的队头提取烂橘子坐标,然后分别看它的上下左右四个方向是否为好橘子,如果是则将其传染为烂橘子,并添加到烂橘子队列的队尾,同时其腐烂时间为队头橘子腐烂时间的下一分钟。按照上面的过程直到所有的烂橘子都扩散完成。
接下来的check部分,检查之前记录的好橘子队列,如果其中还存在新鲜橘子,返回-1。否则在好橘子队列检查完后,返回最后一个橘子被腐烂的时间,即经过多少分钟格子中没有新鲜橘子。
参考代码
class Solution {
public:
int orangesRotting(vector<vector<int>>& grid) {
int m = grid.size();
int n = grid[0].size();
int badque[105],badl=0,goodque[105],goodl=0,hang,lie,timee[105];
for(int i=0;i<m;i++)//预处理
{
for(int j=0;j<n;j++)
{
if(grid[i][j]==2)
{
badl++;
badque[badl]=i*100+j;
}
if(grid[i][j]==1)
{
goodl++;
goodque[goodl]=i*100+j;
}
}
}
int t=1,w=badl;
while(t<=w)//扩散
{
hang=badque[t]/100;
lie=badque[t]%100;
if(hang-1>=0&&grid[hang-1][lie]==1)
{
grid[hang-1][lie]=2;
badque[++w]=(hang-1)*100+lie;
timee[w]=timee[t]+1;
}
if(hang+1<m&&grid[hang+1][lie]==1)
{
grid[hang+1][lie]=2;
badque[++w]=(hang+1)*100+lie;
timee[w]=timee[t]+1;
}
if(lie-1>=0&&grid[hang][lie-1]==1)
{
grid[hang][lie-1]=2;
badque[++w]=hang*100+lie-1;
timee[w]=timee[t]+1;
}
if(lie+1<n&&grid[hang][lie+1]==1)
{
grid[hang][lie+1]=2;
badque[++w]=hang*100+lie+1;
timee[w]=timee[t]+1;
}
t++;
}
for(int i=1;i<=goodl;i++)//check
{
hang=goodque[i]/100;
lie=goodque[i]%100;
if(grid[hang][lie]==1)
return -1;
}
return timee[w];
}
};