HDU 2102 A计划（三维BFS）

来源：互联网发布：淘宝上卖酒需要什么编辑：程序博客网时间：2024/05/21 10:56

1.略有难度的三维BFS，存储图的数组需要三维

2.需要考虑多种极端情况，比如两层的对应位置为‘#’或者传送门后就是终点的情况

3.下面附上代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <iostream>
#include <string>
using namespace std;
int row,col,time,num,ex,ey,ef;
string s1[15],s2[15];
char Graph[15][15][2];
int vis[15][15][2];
int moves1[4]={ 0, 0, 1,-1};
int moves2[4]={-1, 1, 0, 0};
struct node{
int x,y,floor,step;
};
void init()
{
memset(vis,0,sizeof(vis));
for(int i=0;i<row;i++){
for(int j=0;j<col;j++){
Graph[i+1][j+1][0]=s1[i].at(j);
if(Graph[i+1][j+1][0]=='P'){
ex=i+1;
ey=j+1;
ef=0;
}
}
}
for(int i=0;i<row;i++){
for(int j=0;j<col;j++){
Graph[i+1][j+1][1]=s2[i].at(j);
if(Graph[i+1][j+1][1]=='P'){
ex=i+1;
ey=j+1;
ef=1;
}
}
}//初始化找到终点的位置
for(int i=0;i<=row+1;i++)
{
Graph[i][0][0]='*';
Graph[i][0][1]='*';
Graph[i][col+1][0]='*';
Graph[i][col+1][1]='*';
}
for(int i=0;i<=col+1;i++)
{
Graph[0][i][0]='*';
Graph[0][i][1]='*';
Graph[row+1][i][0]='*';
Graph[row+1][i][1]='*';
}//初始化，将图的四周加上边界避免出界的判断
}
int bfs(int endx,int endy,int endfloor)
{
queue<node> List;
node start;
start.x=start.y=1;
start.floor=start.step=0;
List.push(start);
int now=start.step;
while(!List.empty())
{
node pos=List.front();
List.pop();
now=pos.step;
if(pos.x==endx&&pos.y==endy&&pos.floor==endfloor){
return now;//在队头若找到终点返回到达所需的最短时间
}
int xx=pos.x,yy=pos.y,ss=pos.floor;
for(int i=0;i<4;i++)
{
if(Graph[moves1[i]+xx][moves2[i]+yy][ss]=='#'
&&vis[moves1[i]+xx][moves2[i]+yy][ss]==0)
{
if((Graph[moves1[i]+xx][moves2[i]+yy][(ss+1)%2]=='.'
||Graph[moves1[i]+xx][moves2[i]+yy][(ss+1)%2]=='P')
&&vis[moves1[i]+xx][moves2[i]+yy][(ss+1)%2]==0)
{
vis[moves1[i]+xx][moves2[i]+yy][ss]=1;
vis[moves1[i]+xx][moves2[i]+yy][(ss+1)%2]=1;
node newnode;
newnode.x=moves1[i]+xx;
newnode.y=moves2[i]+yy;
newnode.floor=(ss+1)%2;
newnode.step=now+1;
List.push(newnode);
}
}
else if(Graph[moves1[i]+xx][moves2[i]+yy][ss]=='.'
&&vis[moves1[i]+xx][moves2[i]+yy][ss]==0)
{
vis[moves1[i]+xx][moves2[i]+yy][ss]=1;
node newnode;
newnode.x=moves1[i]+xx;
newnode.y=moves2[i]+yy;
newnode.step=now+1;
newnode.floor=ss;
List.push(newnode);
}
else if(Graph[moves1[i]+xx][moves2[i]+yy][ss]=='P'
&&vis[moves1[i]+xx][moves2[i]+yy][ss]==0)
{
vis[moves1[i]+xx][moves2[i]+yy][ss]=1;
node newnode;
newnode.x=moves1[i]+xx;
newnode.y=moves2[i]+yy;
newnode.step=now+1;
newnode.floor=ss;
List.push(newnode);
}
}
}
return -1;//如果没有找到返回-1
}
int main()
{
scanf("%d",&num);
while(num--)
{
memset(vis,0,sizeof(vis));
scanf("%d%d%d",&row,&col,&time);
for(int i=0;i<row;i++){
cin>>s1[i];
}
for(int i=0;i<row;i++){
cin>>s2[i];
}
init();
int ans=bfs(ex,ey,ef);
if(ans<=time&&ans>0)//如果可以达到终点并且时间条件允许，则输出yes
printf("YES\n");
else
printf("NO\n");
}
return 0;
}

0 0