DFS(depth first search)深度优先遍历算法模板
PS:这里以一道全排列问题为例子,洛谷P1706 全排列问题,这是一道典型的dfs问题。
ps:原理类似一棵树,从根节点开始,每次遍历都先把左子树(左节点)走到底,走到底后再次返回,查看这一个节点的右子树(右节点),遍历的同时要做好标记(走过的路要做好标记),(回溯)返回的时候要把标记取消掉,相当于后悔了,要重新走另一条路。一直遍历,直到得到我们想要的结果,这个图的遍历过程如下方简图。注意:暴力搜索一般是得部分的分的,数据范围比较小的时候都可以考虑dfs暴力搜索。
代码模板,主要有终止条件,处理节点,回溯撤销结果,三个部分,大但并不是所有的题都是这样的模板,根据题意随机应变
代码实现
ps: mark数组用于标记走过的路,ans存储每一次的结果,n表示输入,u表示遍历的每一层!
#include<bits/stdc++.h>
using namespace std;
const int N = 10;
int ans[N];
bool mark[N];
int n;
void dfs(int u)
{
// u为遍历的层数,i为第几个数,结合mark[i]表示是否被标记
if(u == n){
for(int i=0;i<n;i++) cout << " " << ans[i]; //每个空五格,注意:这里只有四个空格
// 四个空格 = 五个场宽
// 每个数字保留 5个场宽不等于每个数字前有5个空格,比赛最好用printf("%5d",ans[i]);。
// printf("%5d",ans[i]);
puts(""); //换行
return; //输出后退出当前层数,回溯上一层
}
for(int i=1;i<=n;i++)
{
// 如果没被标记,赋值,递归调用dfs(u+1)遍历下一层
if(mark[i] == false)
{
ans[u] = i;
mark[i] = true;
dfs(u+1);
// 复原,每次遍历完一次结果输出后,需要复原,好用ans数组存新的情况
// 复原至关重要,负责只能的到一次结果
mark[i] = false;
ans[u] = 0;
}
}
}
int main()
{
cin >> n;
dfs(0);
return 0;
}
代码执行过程推导
/*
u i mark[i] ans[u]
0 1 true 1 dfs[0]
1 2 true 2 dfs[1]
2 3 true 3 dfs[2]
3 print(1,2,3) dfs[3]
开始回溯
2 3 false 0 dfs[3]-->dfs[2]
1 2 false 0 dfs[2]-->dfs[1]
1 3 true 3 继续执行dfs[1],此时i = 3
2 2 true 2 dfs[2]
3 print(1,3,2) dfs[3]
开始回溯
2 2 false 0 dfs[3]-->dfs[2]
2 3 true 0 dfs[2]继续执行,mark[3] =true,不执行if里面的内容
1 3 false 0 dfs[2]-->dfs[1]
0 1 false 0 dfs[1]-->dfs[0]
0 2 true 2 dfs[0]
1 1 true 1 dfs[1]
2 3 true 3 dfs[2]
3 print(2,1,3) dfs[3]
开始回溯
2 3 false 0 dfs[3]-->dfs[2]
1 1 false 0 dfs[2]-->dfs[1]
1 3 true 3 dfs[1]继续执行,mark[2] =true跳过,执行mark[3],下一步dfs[2]
2 1 true 1 dfs[2]
3 print(2,3,1)
开始回溯.......
后面也是同理,重复上述的遍历过程
*/
参考资料及相关链接