Code Forces 585 B. Phillip and Trains(BFS)

Description
有一个3*n的图,上面有k列火车(用字母表示),一个人开始时在第一排某列处,他每秒有三种不同的走法:1.往右走一步 2.往右走一步再往上走一步 3.往右走一步再往下走一步,每一秒在人走过之后所有火车会向左走两步,问这个人能否成功到达最右边,如果可以输出YES,否则输出NO
Input
第一行为一整数t表示用例组数,每组用例第一行为两个整数n和k表示图的列数和火车数,之后一个3*n矩阵表示这张图
Output
对于每组用例,如果人可以走到最右边则输出YES,否则输出NO
Sample Input
2
16 4
…AAAAA……..
s.BBB……CCCCC
……..DDDDD…
16 4
…AAAAA……..
s.BBB….CCCCC..
…….DDDDD….
Sample Output
YES
NO
Solution
BFS,人每次只有三种走法,火车每次向左走两格相当于人再向右走两个,每次都需要判断走的这两格是否满足条件
Code

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
int T,n,k,s;
char m[4][111];
typedef pair<int,int>P;
bool bfs(int sx)
{
    queue<P> que; 
    que.push(P(sx,1));//起点进队 
    while(que.size()) 
    {
        int x=que.front().first,y=que.front().second;//队首元素出队 
        que.pop();
        if(isalpha(m[x][++y])) continue;//向右走一步遇见火车 
        if(y>=n) return true;//到达最右边
        for(int i=-1;i<=1;i++)//上中下枚举 
        {
            int xx=x+i;
            if(xx<1||xx>3) continue;//出界 
            if(isalpha(m[xx][y])||isalpha(m[xx][y+1])||isalpha(m[xx][y+2])||m[xx][y+2]==1) continue;//不合法情况 
            int yy=y+2;
            if(yy>=n) return true;//到达最右边
            m[xx][yy]=1;//标记 
            que.push(P(xx,yy));//进队 
        }
    }
    return false;
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&k);
        memset(m,0,sizeof(m));
        for(int i=1;i<=3;i++)
        {
            scanf("%s",m[i]+1);
            if(m[i][1]=='s')s=i;//找到起点 
        }
        if(bfs(s))printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}

你可能感兴趣的:(Code Forces 585 B. Phillip and Trains(BFS))