第三章
深度优先搜索(DFS)
排列数字
#include<iostream>
using namespace std;
const int N = 10;
int path[N];//保存序列
int state[N];//数字是否被用过
int n;
void dfs(int u)
{
if(u > n)//数字填完了,输出
{
for(int i = 1; i <= n; i++)//输出方案
cout << path[i] << " ";
cout << endl;
}
for(int i = 1; i <= n; i++)//空位上可以选择的数字为:1 ~ n
{
if(!state[i])//如果数字 i 没有被用过
{
path[u] = i;//放入空位
state[i] = 1;//数字被用,修改状态
dfs(u + 1);//填下一个位
state[i] = 0;//回溯,取出 i
}
}
}
int main()
{
cin >> n;
dfs(1);
}
如何实现回溯操作:
回溯操作在return,return完这个递归就结束了,然后恢复现场,而你看懂没,这个u是一直没变过的,这样也就保证我每次都是从头开始遍历,我函数结束的条件就是for循环结束,也就是说for循环的作用其实就是开个排列的头,举个例子就是我for循环开头1,就从1开始递归下去,知道我把1到n个数都找完,也就是我把位子都填满,这个递归才会结束,但是这个递归还会返回上一层,比如1234返回,然后就1243,比较抽象,这个过程,也就是我填完4的时候,他会到递归到下一个位子,然后再来一遍循环,找到3填上去,然后输出,这个循环结束,然后再找到上面的循环,也就是13开头了,然后再往下走再遍历一遍找到2,然后再往下走…
n 皇后的问题
对角线的解决
include <iostream>
using namespace std;
const int N = 20;
// bool数组用来判断搜索的下一个位置是否可行
// col列,dg对角线,udg反对角线
// g[N][N]用来存路径
int n;
char g[N][N];
bool col[N], dg[N], udg[N];
void dfs(int u) {
// u == n 表示已经搜了n行,故输出这条路径
if (u == n) {
for (int i = 0; i < n; i ++ ) puts(g[i]); // 等价于cout << g[i] << endl;
puts(""); // 换行
return;
}
//对n个位置按行搜索
for (int i = 0; i < n; i ++ )
// 剪枝(对于不满足要求的点,不再继续往下搜索)
// udg[n - u + i],+n是为了保证下标非负
if (!col[i] && !dg[u + i] && !udg[n - u + i]) {
g[u][i] = 'Q';
col[i] = dg[u + i] = udg[n - u + i] = true;
dfs(u + 1);
col[i] = dg[u + i] = udg[n - u + i] = false; // 恢复现场 这步很关键
g[u][i] = '.';
}
}
int main() {
cin >> n;
for (int i = 0; i < n; i ++ )
for (int j = 0; j < n; j ++ )
g[i][j] = '.';
dfs(0);
return 0;
}
。