这道题算是一个搜索的板子题了。多源的宽度优先搜索。
大致思路如下:
1.首先遍历统计初始数组中有多少好橘子和坏橘子
如果没有坏橘子,返回-1;
如果没有好橘子,返回0;
2.当好橘子不为0得时候,开始每轮的循环。时间数加1,然后里面所有坏橘子感染周围的好橘子。结束后判断这一轮感染完好橘子数有没有减,没有返回1.有的话且不为0继续下一轮感染,最后就得到了感染时间
代码如下:
int orangesRotting(int** grid, int gridRowSize, int *gridColSizes) {
int good = 0, bad = 0, t = 0;
for(int i = 0; i < gridRowSize; i++)
for(int j = 0; j < gridColSizes[0]; j++)
{
if (grid[i][j] == 1) good++;
else if(grid[i][j] == 2) bad++;
}
if(!good) return 0;
if(!bad) return -1;
int new_good = good;
while(new_good)
{
t++;
for(int i = 0; i < gridRowSize; i++)
for(int j = 0; j < gridColSizes[0]; j++)
if(grid[i][j] == 2)
{
if(i && grid[i - 1][j] == 1) {grid[i-1][j]= 3; new_good--;}
if(j && grid[i][j - 1] == 1) {grid[i][j-1] = 3; new_good--;}
if(i != gridRowSize - 1 && grid[i + 1][j] == 1)
{grid[i+1][j]= 3; new_good--;}
if(j != gridColSizes[0] - 1 && grid[i][j + 1] == 1)
{grid[i][j+1] = 3; new_good--;}
}
for(int i = 0; i < gridRowSize; i++)
for(int j = 0; j < gridColSizes[0]; j++)
if(grid[i][j] == 3) grid[i][j] = 2;
if(new_good == good) return -1;//没有新的橘子变坏
good = new_good;
}
return t;
}