指南第5章 例题1 UVA 11624 Fire!（迷宫问题【多源BFS】）

题目链接：http://7xjob4.com1.z0.glb.clouddn.com/f08168bf425521a08eac790e3a9ae7e3

中文题意：
你的任务是帮助Joe走出一个大火蔓延的迷宫。Joe每分钟可以走到上下左右4个方向的相邻格之一，而所有着火的格子都会往四周蔓延（即如果某个空格与着火格有公共边，则下一分钟这个空格将着火）。迷宫中有一些障碍格，Joe和火都无法进入。当Joe走到一个迷宫的边界格子时，我们认为他已经出了迷宫。
【输入格式】
输入第一行为数据组数T。每组数据第一行为两个整数R和C（1<=R,C<=1000），以下R行每行C个字符，即迷宫。其中“#”表示墙和障碍物，“.”表示空地，“J”是Joe的初始地位置（也是一个空地），“F”是着火格。每组数据的迷宫中恰好有一个格子是“J”。
【输出格式】
对于每组数据，输出走出迷宫的最短时间（单位：分钟）。如果无法走出迷宫，则输出IMPOSSIBLE。

【分析】
如果没有火，那么本题是一个标准的迷宫问题，可以用BFS解决。加上了火，难度增加了多少呢？其实没多少。注意，火是不会自动熄灭的，因此只要某个格子上在某时刻起火了，以后将一直如此。所以只需要预处理每个格子的起火时间，在BFS扩展结点的时候加一个判断，当达到新结点时该格子没着火才真的把这个新结点加到队列中。
最后需要考虑一下如何求出每个格子的起火时间，其实这也是一个最短路问题，只不过起点不是一个，而是多个（所有的初始着火点）。这只需要在队列初始化时把所有着火点都放进去即可。

两个步骤的时间复杂度均为O(RC)。

下面是AC代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
char a[1005][1005];
int sx,sy,book1[1005][1005],book2[1005][1005];
int n,m,dir[4][2]= {{0,1},{0,-1},{1,0},{-1,0}};
struct node
{
    int x,y,step;
};

void bfs1()//计算每个点着火的时间
{
    node fr,ne;
    queue<node>q;
    memset(book1,-1,sizeof(book1));//存储着火时间，-1时代表没有着火
    for(int i=0; i<n; i++)
    {
        for(int j=0; j<m; j++)
        {
            if(a[i][j]=='F')
            {
                book1[i][j]=0;
                fr.x=i;
                fr.y=j;
                fr.step=0;
                q.push(fr);
            }
        }
    }
    while(!q.empty())
    {
        fr=q.front();
        q.pop();
        for(int i=0; i<4; i++)
        {
            ne.x=fr.x+dir[i][0];
            ne.y=fr.y+dir[i][1];
            if(ne.x>=0&&ne.x<n&&ne.y>=0&&ne.y<m&&book1[ne.x][ne.y]==-1&&a[ne.x][ne.y]!='#')
            {
                ne.step=fr.step+1;
                book1[ne.x][ne.y]=fr.step+1;
                q.push(ne);
            }
        }
    }
}

int bfs2(int sx,int sy)
{
    node fr,ne;
    queue<node>q;
    memset(book2,0,sizeof(book2));
    fr.x=sx;fr.y=sy;
    fr.step=0;
    q.push(fr);
    while(!q.empty())
    {
        fr=q.front();
        q.pop();
        for(int i=0; i<4; i++)
        {
            ne.x=fr.x+dir[i][0];
            ne.y=fr.y+dir[i][1];
            ne.step=fr.step+1;//下面判断结点是否可以加入到队列中
            if((ne.step<book1[ne.x][ne.y]||book1[ne.x][ne.y]==-1)&&ne.x>=0&&ne.x<n&&ne.y>=0&&ne.y<m&&book2[ne.x][ne.y]==0&&a[ne.x][ne.y]=='.')
            {
                book2[ne.x][ne.y]=1;
                q.push(ne);
            }
            if(ne.x<0||ne.x>=n||ne.y<0||ne.y>=m)
            {
                return ne.step;
            }
        }
    }
    return -1;
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        for(int i=0; i<n; i++)
        {
            scanf("%s",&a[i]);
            for(int j=0; j<m; j++)//找出Joe所在的位置
            {
                if(a[i][j]=='J')
                {
                    sx=i;
                    sy=j;
                }
            }
        }
        bfs1();
        int re=bfs2(sx,sy);
        if(re==-1)
        {
            printf("IMPOSSIBLE\n");
        }
        else
        {
            printf("%d\n",re);
        }
    }
    return 0;
}