虽然c++有专门的stack来表示栈,但是由于本题要输出路径,所以选择利用vector来模拟栈,因为它不仅可以像stack一样在栈顶插入(push_back)和弹出(pop_back)元素。而且还支持索引,因此在打印路径时比较方便。
另外,dfs的一般模板是:
dfs(参数){
1.if(到达停止状态){
进行一些操作,比如打印路径
return;
}
2.对一般元素进行操作(一般是标记,然后一个for循环访问它的下一层节点)
3.回溯
}
其中回溯非常细节,大家先看代码,主要的函数有三个第一个是check函数,用于检测新的节点是否合法(包括是否越界,是否已经走过,是否能走),第二个printPath函数,用于在到达终点打印路径,第三个就是dfs函数,用于搜索所有可能的路径。
#include<bits/stdc++.h>
using namespace std;
#define MAX 20
int a[MAX][MAX];
int mark[MAX][MAX];
int m,n;
struct p{
int x,y;
};
int cnt=0;
vector<struct p> path;
const int dx[]={0,-1,0,1};
const int dy[]={-1,0,1,0};
bool check(int newx,int newy){
if(newx<1||newx>m||newy<1||newy>n||mark[newx][newy]||a[newx][newy]==0)
return false;
return true;
}
void printPath(int xx,int yy){
for(int i=0;i<path.size();i++){
cout<<"("<<path[i].x<<","<<path[i].y<<")";
cout<<"->";
}
cout<<"("<<xx<<","<<yy<<")";
cout<<endl;
}
void dfs(int x,int y,int xx,int yy){
if(x==xx&&y==yy){
cnt++;
printPath(xx,yy);
return;
}
struct p tmp;
tmp.x=x;
tmp.y=y;
path.push_back(tmp);
mark[x][y]=1;
for(int k=0;k<4;k++){
int newx=x+dx[k];
int newy=y+dy[k];
if(check(newx,newy)){
dfs(newx,newy,xx,yy);
mark[newx][newy]=0;//回溯
//path.pop_back();
}
}
path.pop_back();//访问完这个节点的所有的下一个状态,再弹出这个节点
}
int main(void){
cin>>m>>n;
for(int i=1;i<=m;i++){
for(int j=1;j<=n;j++){
cin>>a[i][j];
}
}
int x,y,xx,yy;
cin>>x>>y;
cin>>xx>>yy;
dfs(x,y,xx,yy);
if(cnt==0) cout<<"-1";
}
其中pop_back必须要在遍历完所有子节点之后进行,可能有人会有疑问,不是应该每次搜索完子节点之后马上回溯吗(。大家不妨像一下,什么时候dfs开始回溯?是不是搜到的子节点是终点或者不合法,dfs才会进行回溯,当搜到终点时,终点压根没入栈,如果这个时候pop_back(),会导致弹出上一层节点,不合法的同理。所以必须在遍历完所有子节点之后再弹出。如果还是不清楚,可以自己举个简单的例子模拟,看看pop_back放在哪个位置才是正确的。