引言
之前刚学DFS的时候并不完全理解为什么递归可以一直往下做,后来直到了递归的本质是栈,就想着能不能手写栈来代替递归呢。当时刚学,自己觉得水平不够就搁置了这个想法,今天上数据结构老师正好讲了栈的应用,其中就有一个走迷宫问题,于是写下这篇文章,希望能帮助大家更好的理解DFS。
B3625 迷宫寻路 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)
DFS
#include<bits/stdc++.h>
const int N=110;
char g[N][N];
bool st[N][N];
int n,m;
int dx[]={0,1,0,-1};
int dy[]={1,0,-1,0};
int flag=0;
void dfs(int x,int y)
{
if(flag) return;
if(x==n&&y==m)
{
flag=1;
return ;
}
for(int i=0;i<4;i++)
{
int a=x+dx[i];
int b=y+dy[i];
if(a<1||b<1||a>n||b>m) continue;
if(g[a][b]=='#') continue;
if(st[a][b]) continue;
st[a][b]=true;
dfs(a,b);
if(flag) return;
st[a][b]=false;
}
return ;
}
signed main()
{
std::cin>>n>>m;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
std::cin>>g[i][j];
}
}
st[1][1]=true;
dfs(1,1);
if(!flag) std::cout<<"No"<<'\n';
else std::cout<<"Yes"<<'\n';
return 0;
}
因为这题数据是100,所以DFS是过不了哒。正解应该是BFS。
栈
栈的写法可以直接ac,效率可见一斑。
#include<bits/stdc++.h>
const int N=110;
typedef std::pair<int,int> PII;
char g[N][N];
bool st[N][N];
int n,m;
int dx[]={0,1,0,-1};
int dy[]={1,0,-1,0};
int flag=0;
void dfs(int x,int y)
{
std::stack<PII> stk;
st[x][y]=true;
stk.push({x,y});
while(!stk.empty())
{
auto t=stk.top();
int a=t.first;
int b=t.second;
if(a==n&&b==m)
{
flag=1;
return ;
}
int ok=0;
for(int i=0;i<4;i++)
{
int na=a+dx[i],nb=b+dy[i];
if(g[na][nb]=='#') continue;
if(st[na][nb]) continue;
if(a<1||b<1||a>n||b>m) continue;
//这个点可以走
ok=1;
st[na][nb]=true;
stk.push({na,nb});
}
if(!ok)
{//不回溯是因为到这一步说明这个点是死胡同
//st[stk.top().first][stk.top().second]=0;
stk.pop();
}
}
return ;
}
signed main()
{
std::cin>>n>>m;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
std::cin>>g[i][j];
}
}
dfs(1,1);
if(flag) std::cout<<"Yes"<<'\n';
else std::cout<<"No"<<'\n';
return 0;
}
BFS
宽度优先搜索
#include<bits/stdc++.h>
typedef std::pair<int,int> PII;
const int N=110;
int n,m;
char g[N][N];
int dist[N][N];
PII q[N*N];
int hh=0,tt=-1;
int dx[]={0,1,0,-1};
int dy[]={1,0,-1,0};
void bfs(int x,int y)
{
memset(dist,-1,sizeof dist);
dist[x][y]=0;
q[++tt]={x,y};
while(hh<=tt)
{
PII t=q[hh++];
for(int i=0;i<4;i++)
{
int a=t.first+dx[i];
int b=t.second+dy[i];
if(dist[a][b]!=-1) continue;
if(g[a][b]=='#') continue;
if(a<1||b<1||a>n||b>m) continue;
q[++tt]={a,b};
dist[a][b]=dist[x][y]+1;
if(a==n&&b==m)
{
std::cout<<"Yes";
return ;
}
}
}
std::cout<<"No";
return ;
}
signed main()
{
std::cin>>n>>m;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
std::cin>>g[i][j];
}
}
bfs(1,1);
return 0;
}