POJ2251 BFS基础题

来源:互联网 发布:牛仔外套品牌推荐知乎 编辑:程序博客网 时间:2024/06/04 23:21

很简单。

失误的是刚开始用DFS,超时了。

#include<iostream>

#include<queue>
using namespace std;
const int N=35;
struct node
{
int l,r,c,step;
node(int a1=0,int a2=0,int a3=0,int ste=0)
{l=a1;r=a2;c=a3;step=ste;}
};
int mov[6][3]={{-1,0,0},{1,0,0},{0,-1,0},{0,1,0},{0,0,-1},{0,0,1}};
char maze[N][N][N];
bool visit[N][N][N];
node start,end1;
int L,R,C;
bool inmaze(int l,int r,int c)
{
if(l>=1&&l<=L&&r>=1&&r<=R&&c>=1&&c<=C&&maze[l][r][c]!='#')
return true;
return false;
}
int ans;
void dfs(int l,int r,int c,int step)
{
if(l==end1.l&&r==end1.r&&c==end1.c)
{
ans=step;
return;
}
for(int i=0;i<6;i++)
{
int _l=l+mov[i][0];
int _r=r+mov[i][1];
int _c=c+mov[i][2];
if(inmaze(_l,_r,_c)&&!visit[_l][_r][_c])
{
visit[_l][_r][_c]=true;
dfs(_l,_r,_c,step+1);
visit[_l][_r][_c]=false;
}
}
}
void bfs()
{
queue<node> que;
que.push(node(start.l,start.r,start.c,0));
visit[start.l][start.r][start.c]=true;
while(!que.empty())
{
node nod=que.front();
que.pop();
int l=nod.l;
int r=nod.r;
int c=nod.c;
if(l==end1.l&&r==end1.r&&c==end1.c)
{
ans=nod.step;
return;
}
//if(que.empty())
//return;
for(int i=0;i<6;i++)
{
int _l=l+mov[i][0];
int _r=r+mov[i][1];
int _c=c+mov[i][2];
if(inmaze(_l,_r,_c)&&!visit[_l][_r][_c])
{
que.push(node(_l,_r,_c,nod.step+1));
visit[_l][_r][_c]=true;
}
}
}
}
int main()
{
//freopen("C:\\Users\\wuyanyisb\\Desktop\\1.txt","r",stdin);
while(scanf("%d%d%d",&L,&R,&C),L!=0||R!=0||C!=0)
{
for(int i=1;i<=L;i++)
for(int j=1;j<=R;j++)
{
scanf("%s",maze[i][j]+1);
for(int k=1;k<=C;k++)
{
if(maze[i][j][k]=='S')
start=node(i,j,k);
else if(maze[i][j][k]=='E')
end1=node(i,j,k);
}
}
ans=0;
memset(visit,0,sizeof(visit));
bfs();
if(ans==0)
printf("Trapped!\n");
else
printf("Escaped in %d minute(s).\n",ans);
}


system("pause");
return 0;
}
原创粉丝点击