UVA 11624 Fire（多源bfs）

#include<iostream>
#include<cstdlib>
#include<sstream>
#include<cstdio>
#include<stack>
#include<cstdio>
#include<map>
#include<set>
#include<queue>
#include<cstring>
#include<cmath>
#include<vector>
#include<math.h>
#include<algorithm>
#include<utility>
#include<bits/stdc++.h>
#define REP(i,a,b) for(int i=a;i<b;i++)
#define REPP(i,a,b) for(int i=a;i<=b;i++)
#define inf 0x3f3f3f3f
using namespace std;
typedef long long ll;
//题意：
//你的任务是帮助Joe走出一个大火蔓延的迷宫。Joe每分钟可以走到上下左右
//4个方向的相邻格之一，而所有着火的格子都会往四周蔓延(即如果某个空格
//与着火格有公共边，则下一分钟这个空格将着火)。迷宫中有一些障碍格
//Joe和火都无法进入。当Joe走到一个迷宫的边界格子时，我们认为他已经走出了迷宫
const int maxn=1005;
char a[maxn][maxn];
int sx,sy,book1[maxn][maxn],book2[maxn][maxn];
int n,m;
int 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++)
            {
                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;
}