2021-bfsTRN-K

https://vjudge.net/contest/425322#problem/K

原题复制粘贴

可怜的公主在一次次被魔王掳走一次次被骑士们救回来之后，而今，不幸的她再一次面临生命的考验。魔王已经发出消息说将在T时刻吃掉公主，因为他听信谣言说吃公主的肉也能长生不老。年迈的国王正是心急如焚，告招天下勇士来拯救公主。不过公主早已习以为常，她深信智勇的骑士LJ肯定能将她救出。
现据密探所报，公主被关在一个两层的迷宫里，迷宫的入口是S（0，0，0），公主的位置用P表示，时空传输机用#表示，墙用*表示，平地用.表示。骑士们一进入时空传输机就会被转到另一层的相对位置，但如果被转到的位置是墙的话，那骑士们就会被撞死。骑士们在一层中只能前后左右移动，每移动一格花1时刻。层间的移动只能通过时空传输机，且不需要任何时间。

思路

双层dfs/bfs
存图肯定用三维比较好，注意，第三维（z方向）只有两层，用0/1表示（马上就会发现巧妙之处）
注意：如果遇到时空穿梭机，穿梭到另一层还是时空穿梭机，会陷入死循环，这种是不能走的。
那么，正常情况下，如果换层，只要异或1就好（毕竟只有0/1）

代码

bfs

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <math.h>
#include <queue>

using namespace std;
int n,m,t;
char s[15][15][3];
int flag=0;
int dir[4][2]={{-1,0},{1,0},{0,1},{0,-1}};
int vis[15][15][3];
struct node
{
	int x,y,z,step;
};

void bfs()
{
    queue<node>q;
    node now;
    now.x=1,now.y=1,now.z=0,now.step=0;
	vis[now.x][now.y][now.z]=1;
	q.push(now);
	int i;
    while(!q.empty())
	{
		now=q.front();
		q.pop();
		if(now.step>t) continue;
		if(s[now.x][now.y][now.z]=='P')
        {
            flag=1;
            while(!q.empty()) q.pop();
            return;
        }
        for(i=0;i<4;i++)
        {
            int nx=now.x+dir[i][0],ny=now.y+dir[i][1];
            if(nx>=1&&nx<=n&&ny>=1&&ny<=m)
            {
                int nz=(s[nx][ny][now.z]=='#'?now.z^1:now.z);
                if(!vis[nx][ny][nz]&&s[nx][ny][nz]!='*')
                {
                    if(s[nx][ny][nz^1]=='#'&&s[nx][ny][nz]=='#') continue;
                    node next;
                    next.x=nx,next.y=ny,next.z=nz,next.step=now.step+1;
                    vis[next.x][next.y][next.z]=1;
                    q.push(next);
                }

            }
        }
	}
}
int main()
{
    int c;
    scanf("%d",&c);
    while(c--)
    {
        flag=0;
        memset(vis,0,sizeof(vis));
        scanf("%d%d%d",&n,&m,&t);
        int i,j,k;
        for(k=0;k<2;k++)
        {
            for(i=1;i<=n;i++)
            {
                for(j=1;j<=m;j++)
                {
                    cin>>s[i][j][k];
                }
            }
        }
        bfs();
        if(!flag)
            printf("NO\n");
        else printf("YES\n");
    }
    return 0;
}
/*
1
5 5 14
S*#*.
.#...
.....
****.
...#.

..*.P
#.*..
***..
...*.
*.#..
*/

dfs

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <math.h>

using namespace std;
int n,m,t,endx,endy,endz;
char s[15][15][3];
int flag=0;
int dir[4][2]={{-1,0},{1,0},{0,1},{0,-1}};
int vis[15][15][3];

void dfs(int x,int y,int z,int step)
{
    vis[x][y][z]=1;
    if(x==endx&&y==endy&&z==endz&&step<=t)
    {
        flag=1;
        return;
    }
    if(flag==1) return;
    if(step>t) return;
    int i;
    for(i=0;i<4;i++)
    {
        int nx=x+dir[i][0],ny=y+dir[i][1];
        if(nx>=1&&nx<=n&&ny>=1&&ny<=m)
        {
            int nz=(s[nx][ny][z]=='#'?z^1:z);
            if(s[nx][ny][nz]!='*'&&!vis[nx][ny][nz])
            {
                if(s[nx][ny][nz]=='#'&&s[nx][ny][nz^1]=='#') continue;
                dfs(nx,ny,nz,step+1);
                if(flag==1) return;
                vis[nx][ny][nz]=0;
            }
        }
    }

}
int main()
{
    int c;
    scanf("%d",&c);
    while(c--)
    {
        flag=0;
        memset(vis,0,sizeof(vis));
        scanf("%d%d%d",&n,&m,&t);
        int i,j,k;
        for(k=0;k<2;k++)
        {
            for(i=1;i<=n;i++)
            {
                for(j=1;j<=m;j++)
                {
                    cin>>s[i][j][k];
                    if(s[i][j][k]=='P')
                    {
                        endx=i,endy=j,endz=k;
                    }
                }
            }
        }
        dfs(1,1,0,0);
        if(!flag)
            printf("NO\n");
        else printf("YES\n");
    }
    return 0;
}