C++知识点总结(37)：回溯算法

一、DFS

深度优先搜索，简称 DFS，需要用到基础的两种算法：

• 递归算法：实现了深度优先搜索的基本思想：不撞南墙不回头。
• 回溯算法：实现了深度优先搜索拔 flag 过程，使用逐步试探和撤销的方式来查找问题的解，适用于穷举、搜索。

二、排列例题

1. 全排列

1.1 审题

题目描述

题目名称：Permutations
从键盘读入一个整数 $n$，请输出 $1-n$ 中所有整数的全排列，按照由构成的字典序从小到大输出结果，每组的 $n$ 个数之间用空格隔开。
全排列的含义：从 $n$ 个不同元素中任取 $m$（$m\le n$）个元素，按照一定的顺序排列起来，叫做从 $n$ 个不同元素中取出 $m$ 个元素的一个排列。当 $m=n$ 时所有的排列情况叫全排列。
如当 $n=3$ 时，全排列的结果为：

1 2 3
1 3 2
2 1 3
2 3 1
3 1 2
3 2 1

输入描述

输入文件：Permutations.in
总共输入一个整数 $n$（$1\le n\le 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$ 中选择满足条件的数字，存储到对应的位置上。

Ⅱ 伪代码

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$ 的 $1$ 到 $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$ 个互不相等的数中，选出m个数的组合，请问有哪些不同的选法，每组数输出时要求按照从小到大的顺序输出。比如，假设有 $5$ 个数分别是 1 2 3 4 5，从中选出 $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$ 和 $m$
表示从 $1$ 到 $n$ 中要进行 $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\le n\le 10$

2. 思路

我们可以根据 【二、1.】 来修正代码。我们需要增加 $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\times 6$ 的跳棋棋盘，有六个棋子被放置在棋盘上，使得每行、每列有且只有一个，每条对角线（包括两条主对角线的所有平行线）上至多有一个棋子。

上面的布局可以用序列 $246135$ 来描述，第 $i$ 个数字表示在第 $i$ 行的相应位置有一个棋子，如下：
行号 $123456$
列号 $246135$
这只是棋子放置的一个解。请编一个程序找出所有棋子放置的解。
并把它们以上面的序列方法输出，解按字典顺序排列。
请输出前 $3$ 个解。最后一行是解的总个数。

输入格式

一行一个正整数 $n$，表示棋盘是 $n\times n$ 大小的。

输出格式

前三行为前三个解，每个解的两个数字之间用一个空格隔开。第四行只有一个数字，表示解的总数。

样例1

输入

6

输出

2 4 6 1 3 5
3 6 2 5 1 
4 1 5 2 6 3
4

提示

【数据范围】
对于 $100\%$ 的数据，$6\le n\le 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;
}