回溯法在我们解题步骤中经常被提到,这也是一种常用的方法,回溯法是一种经常被用在 深度优先搜索(DFS)和广度优先搜索(BFS)的技巧。其本质是:走不通就回头。本篇将结合经典例题帮助大家对回溯法的理解。
一、工作原理:
(1)构造空间树;
(2)进行遍历;
(3)如遇到边界条件,即不再向下搜索,转而搜索另一条链;
(4)达到目标条件,输出结果。
二、经典例题
例题一:0-1背包问题
问题:给定n种物品和一背包。物品i的重量是wi,其价值为pi,背包的容量为C。问应如何选择装入背包的物品,使得装入背包中物品的总价值最大?
分析:问题是n个物品中选择部分物品,可知,问题的解空间是子集树。比如物品数目n=3时,其解空间树如下图,边为1代表选择该物品,边为0代表不选择该物品。使用x[i]表示物品i是否放入背包,x[i]=0表示不放,x[i]=1表示放入。回溯搜索过程,如果来到了叶子节点,表示一条搜索路径结束,如果该路径上存在更优的解,则保存下来。如果不是叶子节点,是中点的节点(如B),就遍历其子节点(D和E),如果子节点满足剪枝条件,就继续回溯搜索子节点。

代码如下:
#include <stdio.h>
#define N 3 //物品的数量
#define C 16 //背包的容量
int w[N]={10,8,5}; //每个物品的重量
int v[N]={5,4,1}; //每个物品的价值
int x[N]={0,0,0}; //x[i]=1代表物品i放入背包,0代表不放入
int CurWeight = 0; //当前放入背包的物品总重量
int CurValue = 0; //当前放入背包的物品总价值
int BestValue = 0; //最优值;当前的最大价值,初始化为0
int BestX[N]; //最优解;BestX[i]=1代表物品i放入背包,0代表不放入
//t = 0 to N-1
void backtrack(int t)
{
//叶子节点,输出结果
if(t>N-1)
{
//如果找到了一个更优的解
if(CurValue>BestValue)
{
//保存更优的值和解
BestValue = CurValue;
for(int i=0;i<N;++i) BestX[i] = x[i];
}
}
else
{
//遍历当前节点的子节点:0 不放入背包,1放入背包
for(int i=0;i<=1;++i)
{
x[t]=i;
if(i==0) //不放入背包
{
backtrack(t+1);
}
else //放入背包
{
//约束条件:放的下
if((CurWeight+w[t])<=C)
{
CurWeight += w[t];
CurValue += v[t];
backtrack(t+1);
CurWeight -= w[t];
CurValue -= v[t];
}
}
}
//PS:上述代码为了更符合递归回溯的范式,并不够简洁
}
}
int main(int argc, char* argv[])
{
backtrack(0);
printf("最优值:%d\n",BestValue);
for(int i=0;i<N;i++)
{
printf("最优解:%-3d",BestX[i]);
}
return 0;
}例题二:描述N皇后问题
问题:在n×n格的棋盘上放置彼此不受攻击的n个皇后。按照国际象棋的规则,皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。
N皇后问题等价于在n×n格的棋盘上放置n个皇后,任何2个皇后不放在同一行或同一列或同一斜线上。
分析:从n×n个格子中选择n个格子摆放皇后。可见解空间树为子集树。
使用Board[N][N]来表示棋盘,Board[i][j]=0 表示(I,j)位置为空,Board[i][j]=1 表示(I,j)位置摆放有一个皇后。
全局变量way表示总共的摆放方法数目。
使用Queen(t)来摆放第t个皇后。Queen(t) 函数符合子集树时的递归回溯范式。当t>N时,说明所有皇后都已经摆 放完成,这是一个可行的摆放方法,输出结果;否则,遍历棋盘,找皇后t所有可行的摆放位置,Feasible(i,j) 判断皇后t能否摆放在位置(i,j)处,如果可以摆放则继续递归摆放皇后t+1,如果不能摆放,则判断下一个位置。
Feasible(row,col)函数首先判断位置(row,col)是否合法,继而判断(row,col)处是否已有皇后,有则冲突,返回0,无则继续判断行、列、斜方向是否冲突。斜方向分为左上角、左下角、右上角、右下角四个方向,每次从(row,col)向四个方向延伸一个格子,判断是否冲突。如果所有方向都没有冲突,则返回1,表示此位置可以摆放一个皇后。

代码如下:
#include <stdio.h>
#define N 8
int Board[N][N];//棋盘 0表示空白 1表示有皇后
int way;//摆放的方法数
//判断能否在(x,y)的位置摆放一个皇后;0不可以,1可以
int Feasible(int row,int col)
{
//位置不合法
if(row>N || row<0 || col >N || col<0)
return 0;
//该位置已经有皇后了,不能
if(Board[row][col] != 0)
{ //在行列冲突判断中也包含了该判断,单独提出来为了提高效率
return 0;
}
//
//下面判断是否和已有的冲突
//行和列是否冲突
for(int i=0;i<N;++i)
{
if(Board[row][i] != 0 || Board[i][col]!=0)
return 0;
}
//斜线方向冲突
for(int i=1;i<N;++i)
{
/* i表示从当前点(row,col)向四个斜方向扩展的长度
左上角 \ / 右上角 i=2
\/ i=1
/\ i=1
左下角 / \ 右下角 i=2
*/
//左上角
if((row-i)>=0 && (col-i)>=0) //位置合法
{
if(Board[row-i][col-i] != 0)//此处已有皇后,冲突
return 0;
}
//左下角
if((row+i)<N && (col-i)>=0)
{
if(Board[row+i][col-i] != 0)
return 0;
}
//右上角
if((row-i)>=0 && (col+i)<N)
{
if(Board[row-i][col+i] != 0)
return 0;
}
//右下角
if((row+i)<N && (col+i)<N)
{
if(Board[row+i][col+i] != 0)
return 0;
}
}
return 1; //不会发生冲突,返回1
}
//摆放第t个皇后 ;从1开始
void Queen(int t)
{
//摆放完成,输出结果
if(t>N)
{
way++;
/*如果N较大,输出结果会很慢;N较小时,可以用下面代码输出结果
for(int i=0;i<N;++i){
for(int j=0;j<N;++j)
printf("%-3d",Board[i][j]);
printf("\n");
}
printf("\n------------------------\n\n");
*/
}
else
{
for(int i=0;i<N;++i)
{
for(int j=0;j<N;++j)
{
//(i,j)位置可以摆放皇后,不冲突
if(Feasible(i,j))
{
Board[i][j] = 1; //摆放皇后t
Queen(t+1); //递归摆放皇后t+1
Board[i][j] = 0; //恢复
}
}
}
}
}
//返回num的阶乘,num!
int factorial(int num)
{
if(num==0 || num==1)
return 1;
return num*factorial(num-1);
}
int main(int argc, char* argv[])
{
//初始化
for(int i=0;i<N;++i)
{
for(int j=0;j<N;++j)
{
Board[i][j]=0;
}
}
way = 0;
Queen(1); //从第1个皇后开始摆放
//如果每个皇后都不同
printf("考虑每个皇后都不同,摆放方法:%d\n",way);//N=8时, way=3709440 种
//如果每个皇后都一样,那么需要除以 N!出去重复的答案(因为相同,则每个皇后可任意调换位置)
printf("考虑每个皇后都不同,摆放方法:%d\n",way/factorial(N));//N=8时, way=3709440/8! = 92种
return 0;
}PS:该问题还有更优的解法。充分利用问题隐藏的约束条件:每个皇后必然在不同的行(列),每个行(列)必然也只有一个皇后。这样我们就可以把N个皇后放到N个行中,使用Pos[i]表示皇后i在i行中的位置(也就是列号)(i = 0 to N-1)。这样代码会大大的简洁,因为节点的子节点数目会减少,判断冲突也更简单。
| 1460 | 蓝桥杯基础练习VIP-2n皇后问题 |
C语言网提供由在职研发工程师或ACM蓝桥杯竞赛优秀选手录制的视频教程,并配有习题和答疑,点击了解:
一点编程也不会写的:零基础C语言学练课程
解决困扰你多年的C语言疑难杂症特性的C语言进阶课程
从零到写出一个爬虫的Python编程课程
只会语法写不出代码?手把手带你写100个编程真题的编程百练课程
信息学奥赛或C++选手的 必学C++课程
蓝桥杯ACM、信息学奥赛的必学课程:算法竞赛课入门课程
手把手讲解近五年真题的蓝桥杯辅导课程