CHAPTER_4 算法初步入门
4.3 递归
题目1:
一般把1~N这N个数字按某个顺序摆放的结果称为这N个数的一个排列,这N个数所能形成的所有排列称为它的全排列。如对于1、2两个数字,全排列为(1,2)、(2,1)。
现在给出N个数字,按字典序输出它的全排列。
(字典序:对于两个排列A=(a1,a2,...,an) B=(b1,b2,...,bn),当存在一个i使a1=b1,a2=b2,...,ai-1=bi-1且ai<bi时,字典序A<B)
思路:
从递归的角度考虑,将其分解为若干子问题。
先设置一个足够长的数组P[maxn]用来存放排列结果,用以排列完成后输出结果。从第1位到第n位往P[max]中依次填入数字模拟排列过程。假设当前已经填好了P[1]~P[index-1],正要填写第index位,我们枚举1~n中每个数字并判断其在前面index-1位中是否出现,如果没有出现则填入当前位置,然后进入下一位填写。
上面思路中,我们还有一个问题没有解决:如何判断当前枚举数字在前面index-1位中是否出现过?因此我们设置一个哈希数组hashtable[n],将所有值初始化为0。成功将枚举数字x填入当前位时,令hashtable[x]=1,表示该数字已经出现,这样在后面位置的填写中我们得以判断x是否出现。
参考代码:
#include<iostream>
using namespace std;
const int maxn=15;
int n,p[maxn],hashtable[maxn]={0};
//P存放排列,hashtable用于判断
void current_index(int index){
if(index==n+1){
for(int i=1;i<=n;i++){
cout<<p[i];
if(i==n)
cout<<endl;
else
cout<<' ';
}
return; //递归边界,当index为n+1时表示n位已经排完,控制格式输出一种结果,然后退出
}
for(int x=1;x<=n;x++){
if(hashtable[x]==0){ //如果x还没有被存放
p[index]=x; //将x存放于当前index位置
hashtable[x]=1; //表示x已存放,用于后面位置判断
current_index(index+1); //处理后面一位、后面两位...直至处理至n位
hashtable[x]=0; //后续子问题处理完,还原
}
}
}
int main(){
cin>>n;
current_index(1);
return 0;
}
题目2:
N皇后问题——在n×n格的国际象棋上摆放n个皇后,使其不能互相攻击,即任意两个皇后都不能处于同一行、同一列或同一斜线上,问有多少种摆法?
思路:从全排列的角度来考虑。对于每个皇后,他们必须在不同的列、不同的行。那么将n列皇后所对应的所有行号可能写出来,就是一个规模为n的全排列问题。因此枚举每一个排列,判断该排列中是否有皇后处在一个斜线上,若为否,则获得一种符合要求摆法。
优化思路:在上述思路中可以发现,有时放置了部分皇后时,剩余的皇后无论怎么放都不符合要求,此时可以return返回上层,进而减少计算量。这种做法称作回溯。
参考代码:
#include<iostream>
#include<cmath>
using namespace std;
const int maxn=15;
int n,count=0,p[maxn],hashtable[maxn]={0};
void current_index(int index){ //该函数表示处理第index列皇后的放置 ,p[index]表示放置的行号
if(index==n+1){
count++;
return; //经过下面的回溯处理后,能达到边界的一定是合法排列 ,计数+1
}
for(int x=1;x<=n;x++){
if(hashtable[x]==0){ //x行没有皇后
int flag=1;
for(int pre=1;pre<index;pre++){ //遍历之前的皇后
if(abs(index-pre)==abs(x-p[pre])){
flag=0;
break; //回溯处理 ,有皇后在一条斜线上时直接返退出
}
}
if(flag){
p[index]=x;
hashtable[x]=1;
current_index(index+1);
hashtable[x]=0;
} //将index列皇后放在x行,哈希表置1表示该行已有皇后,处理完子问题后,哈希表置回0
}
}
}
int main(){
cin>>n;
current_index(1);
cout<<count;
return 0;
}