算法思想:
由于题目要求是返回最小完全腐烂的时间,所以,用深度优先搜索很不好做,并且这道题的腐烂传染方式很适合BFS。
1.首先,将开始腐烂的橘子坐标加入队列,遍历这个橘子后,将其上、下、左、右四个方向的橘子入队(下一轮腐烂的橘子)。
2.接着,按照队列的顺序将橘子出队,遍历并将其上、下、左、右四个方向的橘子结点入队。
3.将一轮腐烂的橘子遍历完后,计时+1分钟。
4.重复2、3过程,直到队列为空。
最后,检查是否有新鲜橘子,如果有,返回-1,没有,返回分钟数。
int m, n, ret;//行,列数,时间
int dx[4] = { 0,1,0,-1 };//方向数组
int dy[4] = { 1,0,-1,0 };
int map[1000][1000] = { 0 };
int bfs()
{
queue<pair<int, int>>q;
for (int i = 0; i < m; i++)
{
for (int j = 0; j < n; j++)
{
if (map[i][j] == 2)
q.push(make_pair(i, j));//让坏橘子的坐标入队
}
}
while (!q.empty())
{
int sz = q.size();
bool flag = false;//判断是否有橘子腐烂
while (sz--)
{
pair<int, int>pos = q.front();
q.pop();
for (int i = 0; i < 4; i++)//探查四个方向,判断新鲜橘子,将其腐烂并入队
{
int tx = pos.first + dx[i];
int ty = pos.second + dy[i];
if (tx < 0 || ty < 0 || tx >= n || ty >= m)// 判断是否越界
continue;
if (map[tx][ty] == 1)// 本次有橘子腐烂
{
flag = true;
map[tx][ty] = 2;
q.push(make_pair(tx, ty));
}
}
}
if (flag)//本次有橘子腐烂时间才能+1
ret++;
}
for (int i = 0; i < m; i++)// 判断是否有新鲜橘子
{
for (int j = 0; j < n; j++)
{
if (map[i][j] == 1)
return -1;
}
}
return ret;
}
int main()
{
cin >> m >> n;
for (int i = 0; i < m; i++)
{
for (int j = 0; j < n; j++)
{
cin >> map[i][j];
}
}
cout << bfs() << endl;
return 0;
}