DFS-深度搜索 详解

此篇博客为《啊哈算法》学后感，有需求的读者可自行搜索该书电子版

递归解析

引入思考：如何实现1~N个数字的全排列(例：1-3全排列为 1 2 3，1 3 2，2 1 3，2 3 1，3 1 2，3 2 1)

基本原理

1. 一条路走到黑，按照一种情况完成所有选择
2. 已经走到黑了，所有选择都选完了，此时就要倒退一步，到上一步做和之前在上一步时的不同选择来继续走到黑。

带入题意

以N=3为例，全排列1，2，3三个数

1. 首先我们一条路走到黑，就按照数字的顺序
   第一步的时候我们有三个数字，按照数字的顺序，第一个就选择1。到第二步的时候，就只剩下2和3两个数字了，按照数字的顺序，第二步就选择2。第三步的时候就只剩一个数字了，就只能选择3。此时所有的选择就都做完了，就是走到黑。
   
2. 按照上一次走到黑的路，我们只返回前面一步，然后重新做选择
   从上面的第三步走完，我们已经是完整了走完了一条路，此时就要倒退一步，退回到上面的第二步，然后做和上次在第二步时做的不同的选择，即上一次第二步我们剩两个数字时我们选择了2，那么这一次我们就做不一样的选择，我们就选择3。随后第三步则依旧按照上面1叙述的方法继续走。
   
3. 每一次都倒退回前一步，改变了一次选择，故就算剩下的所有选择都按照1叙述的方法继续走，也不会有重复的数字出现
   比如现在已经排完了123和132，在确定第一步是1的情况下，第二步可以做的选择我们已经做完了，此时就要改变第二步的前一步，也就是改变第一步的选择，这个选择我们依旧按照数字顺序，我们选择2，那么剩下的两个数1和3，则会按照上面一样的方法走完所有选择

将思路转换为代码

1. 首先我们一条路走到黑，就按照数字的顺序

// 定义一个数组来存储这些数字
int a[3];
//按照数字的顺序,为第step步，选择一个数字
for(int i=1;i<=n;i++)	a[step]=i; 

2. 如何走到黑时返回到上一步？->将其封装

int a[3],book[3]
//step为第几步
void dfs(int step){
	if(step==n+1){ 
	//step=n+1时，代表已经走完了n步，即前面的n个数都已经选择完了
	return ;	//已经走到黑了，就需要退回到上一步
	}
	else{
		//未都走到黑，我们继续走，做出这一步的选择
		for(int i=1;i<=n;i++)	a[step]=i; 
		
	}
}

3. 返回到前面一步，如何做出不同的选择？->增加一个数组来标记上一步的选择

// 我们可以在引入一个数组book[3],来标记数有无被选择过
int a[3],book[3]
//step为第几步
void dfs(int step){
	if(step==n+1){ 
	//step=n+1时，代表已经走完了n步，即前面的n个数都已经选择完了
	return ;	//已经走到黑了，就需要退回到上一步
	}
	else{
		//未都走到黑，我们继续走，做出这一步的选择
		for(int i=1;i<=n;i++){
				if(book[i]==0){
				//该数未被选择
				a[step]=i;
				//被选择过的数做标记
				book[i]=1;
				//这一步选择完了，我们继续下一步选择
				dfs(step+1);
				//注意，上面的dfs会在step=n+1时返回到这一步，代表下面所有的选择都已经做完
				//此时就需要改变下面的上一步，也就是现在这一步的选择，我们之前是选择的数字i，并做了标记，为了做不一样的选择，此时的标记就应该去掉，代表我们不做这个选择。
				book[i]=0;
			}
		}
	}
}

完整代码

#include<bits/stdc++.h>
using namespace std;
int n;
int a[3],book[3];
//step为第几步
void dfs(int step){
	if(step==n+1){ 
	//step=n+1时，代表已经走完了n步，即前面的n个数都已经选择完了
	for(int i=1;i<=n;i++) cout<<a[i]<<" ";
	cout<<endl;
	return ;	//已经走到黑了，就需要退回到上一步
	}
	else{
		//未都走到黑，我们继续走，做出这一步的选择
		for(int i=1;i<=n;i++){
				if(book[i]==0){
				//该数未被选择
				a[step]=i;
				//被选择过的数做标记
				book[i]=1;
				//这一步选择完了，我们继续下一步选择
				dfs(step+1);
				//注意，上面的dfs会在step=n+1时返回到这一步，代表下面所有的选择都已经做完
				//此时就需要改变下面的上一步，也就是现在这一步的选择，我们之前是选择的数字i，并做了标记，为了做不一样的选择，此时的标记就应该去掉，代表我们不做这个选择。
				book[i]=0;
			}
		}
	}
}
int main(){
	cout<<"输入数字n："; 
	cin>>n;
	dfs(1); //从第一步开始走 
	return 0;
}

结果显示

注意,我编写代码的时候，数组只开了3，要是全排列更大的数，数组空间应该相应变大