回溯算法
一、DFS
深度优先搜索,简称
DFS
,需要用到基础的两种算法:
- 递归算法:实现了深度优先搜索的基本思想:不撞南墙不回头。
- 回溯算法:实现了深度优先搜索拔
flag
过程,使用逐步试探和撤销的方式来查找问题的解,适用于穷举、搜索。
二、排列例题
1. 全排列
1.1 审题
题目描述
题目名称:
Permutations
从键盘读入一个整数 n n n,请输出 1 − n 1-n 1−n 中所有整数的全排列,按照由构成的字典序从小到大输出结果,每组的 n n n 个数之间用空格隔开。
全排列的含义:从 n n n 个不同元素中任取 m m m( m ≤ n m≤n m≤n)个元素,按照一定的顺序排列起来,叫做从 n n n 个不同元素中取出 m m m 个元素的一个排列。当 m = n m=n m=n 时所有的排列情况叫全排列。
如当 n = 3 n=3 n=3 时,全排列的结果为:1 2 3 1 3 2 2 1 3 2 3 1 3 1 2 3 2 1
输入描述
输入文件:
Permutations.in
总共输入一个整数 n n n( 1 ≤ n ≤ 6 1≤n≤6 1≤n≤6)
输出描述
输出文件:
Permutations.out
前若干行每行一个数据,表示全排列的结果,所有全排列按照由小到大输出
最后一行一个整数,表示全排列的个数
样例1
输入
3
输出
1 2 3 1 3 2 2 1 3 2 3 1 3 1 2 3 2 1 6
1.2 思路
Ⅰ 基本思路
每一种方案我们都可以使用数组来存储。
每一次我们都在重复找每个位置存储的数字,思想就是在 1 − n 1-n 1−n 中选择满足条件的数字,存储到对应的位置上。
Ⅱ 伪代码
dfs(pos)
if pos > n
Output solution
return
end if
for i = 1 ~ n
if (vis[i] == 0)
a[pos] = i
dfs(pos+1)
end if
end for
1.3 参考答案
#include <iostream>
#include <cstdio>
using namespace std;
int n;
int cnt;
int a[10];
bool vis[10];
// 找a[pos]里面存什么数字
void dfs(int pos)
{
if (pos > n)
{
// 输出方案
for (int i = 1; i <= n; i++)
{
cout << a[i] << " ";
}
cout << endl;
cnt++; // 方案数增加
return;
}
for (int i = 1; i <= n; i++)
{
if (vis[i] == false) // 数字i未被用过
{
a[pos] = i;
vis[i] = true; // 标记已用
dfs(pos+1); // 递归
vis[i] = 0; // 回溯
}
}
}
int main()
{
freopen("Permutations.in", "r", stdin);
freopen("Permutations.out", "w", stdout);
cin >> n;
dfs(1);
cout << cnt;
fclose(stdin);
fclose(stdout);
return 0;
}
2. 素数环
2.1 审题
给定一个正整数 n n n,要求输出所有长度为 n n n 的 1 1 1 到 n n n 之间的排列成一个环,使得排列中任意两个相邻元素的和为一个质数。
2.2 思路
这道题目和上一题非常类似,其中不同的是还需要判断是否满足素数环的条件。最优的方法就是加在
for-if
的条件中,这样不至于别人告诉你了答案你还硬要写错的,然后越来越错。
2.3 参考答案
#include <iostream>
using namespace std;
int n;
int cnt;
int a[20];
bool vis[20];
bool isPrime(int x)
{
if (x == 1) return false;
for (int i = 2; i * i <= x; i++)
{
if (x % i == 0)
{
return false;
}
}
return true;
}
void dfs(int pos)
{
if (pos > n)
{
if (isPrime(a[1]+a(n)) // 首尾之和是否为质数
{
// 输出方案
for (int i = 1; i <= n; i++)
{
cout << a[i] << " ";
}
cout << endl;
cnt++; // 方案数增加
}
return;
}
for (int i = 1; i <= n; i++)
{
if (vis[i] == false)
{
}
if (pos == 1 || isPrime(i+a[pos-1]))
{
a[pos] = i;
vis[i] = true;
dfs(pos+1);
vis[i] = false;
}
}
}
int main()
{
cin >> n;
dfs(1);
cout << cnt;
return 0;
}
三、组合例题
组合数
1. 审题
题目描述
从 n n n 个互不相等的数中,选出m个数的组合,请问有哪些不同的选法,每组数输出时要求按照从小到大的顺序输出。比如,假设有 5 5 5 个数分别是
1 2 3 4 5
,从中选出 3 3 3 个数的组合有:1 2 3 1 2 4 1 2 5 1 3 4 1 3 5 2 3 4 2 3 5 3 4 5
输入描述
总共输入一行两个整数 n n n 和 m m m
表示从 1 1 1 到 n n n 中要进行 m m m 个数的组合
输出描述
前若干行为从小到大的
样例1
输入
5 3
输出
1 2 3 1 2 4 1 2 5 1 3 4 1 3 5 1 4 5 2 3 4 2 3 5 2 4 5 3 4 5
提示
1 < m ≤ n ≤ 10 1 < m ≤ n ≤10 1<m≤n≤10
2. 思路
我们可以根据 【二、1.】 来修正代码。我们需要增加 m m m 变量,修改一下输出,增加一个判断,使输出都是递增。
3. 参考答案
#include <iostream>
using namespace std;
int n, m;
int a[10];
bool vis[10];
// 找a[pos]里面存什么数字
void dfs(int pos)
{
if (pos > m)
{
// 输出方案
for (int i = 1; i <= m; i++)
{
cout << a[i] << " ";
}
cout << endl;
return;
}
for (int i = a[pos-1]+1; i <= n; i++)
{
if (vis[i] == false) // 数字i未被用过
{
a[pos] = i;
vis[i] = true; // 标记已用
dfs(pos+1); // 递归
vis[i] = 0; // 回溯
}
}
}
int main()
{
cin >> n >> m;
dfs(1);
return 0;
}
四、综合例题
8皇后
1. 审题
题目描述
[USACO1.5] 八皇后 Checker Challenge
一个如下的 6 × 6 6 \times 6 6×6 的跳棋棋盘,有六个棋子被放置在棋盘上,使得每行、每列有且只有一个,每条对角线(包括两条主对角线的所有平行线)上至多有一个棋子。
上面的布局可以用序列 2 4 6 1 3 5 2\ 4\ 6\ 1\ 3\ 5 2 4 6 1 3 5 来描述,第 i i i 个数字表示在第 i i i 行的相应位置有一个棋子,如下:
行号 1 2 3 4 5 6 1\ 2\ 3\ 4\ 5\ 6 1 2 3 4 5 6
列号 2 4 6 1 3 5 2\ 4\ 6\ 1\ 3\ 5 2 4 6 1 3 5
这只是棋子放置的一个解。请编一个程序找出所有棋子放置的解。
并把它们以上面的序列方法输出,解按字典顺序排列。
请输出前 3 3 3 个解。最后一行是解的总个数。
输入格式
一行一个正整数 n n n,表示棋盘是 n × n n \times n n×n 大小的。
输出格式
前三行为前三个解,每个解的两个数字之间用一个空格隔开。第四行只有一个数字,表示解的总数。
样例1
输入
6
输出
2 4 6 1 3 5 3 6 2 5 1 4 1 5 2 6 3 4
提示
【数据范围】
对于 100 % 100\% 100% 的数据, 6 ≤ n ≤ 13 6 \le n \le 13 6≤n≤13。
USACO Training Section 1.5
2. 参考答案
#include <iostream>
using namespace std;
int n;
int cnt;
int a[20];
bool col[20];
bool zd[40], fd[40]; // 正负对角线
void dfs(int row)
{
if (row > n)
{
// 输出方案
if (cnt <= 2)
{
for (int i = 1; i <= n; i++)
{
cout << a[i] << " ";
}
cout << endl;
}
cnt++; // 方案数增加
return;
}
for (int j = 1; j <= n; j++)
{
if (!col[j] && !zd[row+j] && !fz[row-j+n])
{
a[row] = j;
col[j] = true;
zd[row+j] = true;
fd[row-j+n] = true;
dfs(row+1);
col[j] = false;
zd[row+j] = false;
fd[row-j+n] = false;
}
}
}
int main()
{
cin >> n;
dfs(1);
cout << cnt;
return 0;
}