1.题目要求输出路径:用栈来跟踪 DFS 的过程,记录 DFS 走过的路径,是最方便的。
DFS 到某个格子时,把这个格子放到栈里,表示路径增加了这个格子。DFS 回溯的时候,退出了这个格子,表示路径上不再包括这个格子,需要从栈中弹走这个格子。
2.容器vector:vector 既Int型的容器,与数组类似,与数组的区别是容器大小可以变,可以自定增长或缩短空间.
vector的用法请点击进去
vector a;
容器进栈: a.push_back(5); //在a的最后一个向量后插入一个元素,其值为5
容器出栈: a.pop_back(); //删除a向量的最后一个元素
返回元素个数: a.size(); //返回a中元素的个数;
3.dfs函数中分为两部分截至条件和搜索条件…处理第一个点的步骤写在主函数里***重要!!
#include<bits/stdc++.h>
using namespace std;
int n;
int a[25],b[25],vis[25][25];
vector<int> path; //记录路径
int d[4][2]={1,0,-1,0,0,1,0,-1}; //上下左右
void dfs(int x,int y){
if(a[x]<0 || b[y]<0) return; //剪枝
if(x==n-1 && y==n-1){ //走到出口
int ok=1;
for(int i = 0;i < n;i++)
if(a[i]!=0 || b[i]!=0){ //****如果与题目相符的话,最后数组的值为0
ok=0;break;//或者写return
}
if(ok)
for(int i = 0;i < path.size();i++)
cout << path[i] <<" ";
}
for(int i = 0;i < 4;i++){
int tx = x + d[i][0],ty = y + d[i][1];
if(vis[tx][ty]==0 && tx>=0 && tx<n && ty>=0 && ty<n){
vis[tx][ty]=1;
path.push_back(tx*n + ty); //进栈,记录路径
a[tx]--;
b[ty]--;
dfs(tx,ty);
path.pop_back(); //出栈,DFS回溯
a[tx]++;
b[ty]++;
vis[tx][ty]=0;
}
}
}
int main(){
cin >> n;
for(int i = 0;i < n;i++) cin >> b[i];
for(int i = 0;i < n;i++) cin >> a[i];
path.push_back(0);
vis[0][0]=1;
a[0]--;
b[0]--;
dfs(0,0);
return 0;
}