GYM 101755 H.Safe Path（bfs）

Description

给出一个$n\times m$的地图，$.$表示空地，$S$和$F$分别表示起点和终点，$M$表示怪兽，每次只能往上下左右走，一个位置如果到某个怪兽的距离不超过$d$那么走到这个位置就会马上被怪兽吃掉，问是否存在一条安全的最短路径可以从起点到终点

Input

第一行三个整数$n,m,d$表示地图规模和怪兽的攻击范围，之后输入一个$n\times m$的地图

$(2\le n\cdot m\le 2\cdot 10^5,0\le d\le 2\cdot 10^5)$

Output

如果不存在合法路径则输出$-1$，否则输出最短路径长度

Sample Input

5 7 1
S.M…M
…….
…….
M…M..
……F

Sample Output

12

Solution

首先以所有$M$为起点对全图$bfs$求出每个点到$M$的最短距离，如果最短距离不超过$d$则该点不能走，之后对于可以走的点再$bfs$一遍求出终点到起点的最短距离即可，如果不可达则无解

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=200005;
vector<char>s[maxn];
vector<int>dis[maxn],dis2[maxn];
char c[maxn];
int n,m,d,sx,sy,ex,ey;
int dx[]={-1,0,1,0};
int dy[]={0,-1,0,1};
queue<P>que;
int bfs(int x,int y)
{ 
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            dis2[i][j]=INF;
    dis2[x][y]=0;
    if(dis[x][y]>d)que.push(P(x,y));
    while(!que.empty())
    {
        P now=que.front();
        que.pop();
        int x=now.first,y=now.second;
        if(x==ex&&y==ey)break;
        for(int i=0;i<4;i++)
        {
            int xx=x+dx[i],yy=y+dy[i];
            if(xx<1||xx>n||yy<1||yy>m||dis2[xx][yy]!=INF||dis[xx][yy]<=d)continue;
            dis2[xx][yy]=dis2[x][y]+1;
            que.push(P(xx,yy));
        }   
    }
    return dis2[ex][ey];
}
int main()
{
    scanf("%d%d%d",&n,&m,&d);
    for(int i=1;i<=n;i++)
    {
        dis[i].resize(m+1);
        dis2[i].resize(m+1);
        s[i].resize(m+1);
    }
    for(int i=1;i<=n;i++)
    {
        scanf("%s",c+1);
        s[i].push_back(0);
        for(int j=1;j<=m;j++)
        {
            s[i][j]=c[j];
            if(c[j]=='S')sx=i,sy=j;
            if(c[j]=='F')ex=i,ey=j;
        }
    }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        {
            dis[i][j]=INF;
            if(s[i][j]=='M')dis[i][j]=0,que.push(P(i,j));
        }
    while(!que.empty())
    {
        P now=que.front();
        que.pop();
        int x=now.first,y=now.second;
        for(int i=0;i<4;i++)
        {
            int xx=x+dx[i],yy=y+dy[i];
            if(xx<1||xx>n||yy<1||yy>m||dis[xx][yy]!=INF)continue;
            dis[xx][yy]=dis[x][y]+1;
            que.push(P(xx,yy));
        }   
    }
    int ans=bfs(sx,sy);
    if(ans==INF)printf("-1\n");
    else printf("%d\n",ans);
    return 0;
}