- 腐烂的橘子
在给定的 m x n 网格 grid 中,每个单元格可以有以下三个值之一:
值 0 代表空单元格;
值 1 代表新鲜橘子;
值 2 代表腐烂的橘子。
每分钟,腐烂的橘子 周围 4 个方向上相邻 的新鲜橘子都会腐烂。
返回 直到单元格中没有新鲜橘子为止所必须经过的最小分钟数。如果不可能,返回 -1 。
思路
把初始的烂橘子当作第0天感染的,那么接下来一天它可以感染它附近四个方向的橘子,因此四个方向的橘子感染天数就是感染了它的橘子的天数+1。再对这些橘子进行同样的操作。应用队列实现。为了便于判断是不是所有橘子都被感染完了,可以开始统计所有好橘子的数量,感染一个好橘子数量减一。为0的时候全部感染。(这个还是看了官方的代码才用的,我开始是感染了橘子就把grid中改为2,然后遍历有没有1,比较浪费时间)
class Solution {
public:
int orangesRotting(vector<vector<int>>& grid) {
int vi[4]={1,-1,0,0};
int vy[4]={0,0,1,-1};
//pari是一个stl中定义好的结构体,两个成员first,second,
queue<pair<int,int>> qu;
int m=grid.size(),n=grid[0].size();
//result存储橘子坏的时间,假设在-1天所有橘子都是好的
vector<vector<int>> result(m,vector<int>(n,-1));
int cnt=0;
//在第0天,grid中所有初始的坏橘子坏了
for(int i=0;i<m;i++)
{
for(int j=0;j<n;j++)
{
//坏橘子压入队列,同时记录好橘子的数量
if(grid[i][j]==2)
{
qu.push({i,j});
result[i][j]=0;
}else if(grid[i][j]==1)
{
cnt++;
}
}
}
//没有好橘子直接返回0
if(cnt==0)return 0;
int ans=0;
//弹出烂了的橘子,并把它感染的橘子压入队列,新感染的橘子的天数应该是感染它的橘子烂的天数+1
while(!qu.empty())
{
auto q=qu.front();
qu.pop();
int xi=q.first,xy=q.second;
for(int k=0;k<4;k++)
{
int mi=xi+vi[k],ny=xy+vy[k];
//四个方向中,是橘子并且没有被感染过的橘子
if(mi>=0&&mi<m&&ny>=0&&ny<n&&result[mi][ny]==-1&&grid[mi][ny])
{
qu.push({mi,ny});
result[mi][ny]=result[xi][xy]+1;
ans=result[mi][ny];
//每感染一个,好橘子数量减一,数量为0时感染完毕。
cnt--;
if(cnt<=0)
{
return ans;
}
//grid[mi][ny]=2;
}
}
}
// for(int i=0;i<m;i++)
// {
// for(int j=0;j<n;j++)
// {
// if(grid[i][j]==1)
// {
// return -1;
// }
// }
// }
return -1;
}
};