深度优先算法:利用递归函数实现暴力枚举的算法。
递归:指函数自己调用自己本身的过程。
该算法相当于在一个迷宫一直往一个方向,没走到头或者不到终点不停,如果走到这条路的尽头则退回上一个选择点,再次选择不同的方向,直到到达终点为止。
模板
void dfs( int num ){
if( 到达边界 )
{
执行相应的操作
}
for 尝试每一种可能
if 如果满足情况
{
标记
继续下一步 dfs( )
恢复初始状态
}
}
例题
N皇后问题
在N * N的方格棋盘放置了N个皇后,使得它们不相互攻击(即任意2个皇后不允许处在同一排,同一列,也不允许处在与棋盘边框成45角的斜线上。
你的任务是,对于给定的N,求出有多少种合法的放置方法。
输入:共有若干行,每行一个正整数N≤10,表示棋盘和皇后的数量;如果N=0,表示结束。
输出:共有若干行,每行一个正整数,表示对应输入行的皇后的不同放置数量。
ac代码
#include<iostream>
#include<string.h>
#define ios ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
using namespace std;
using pii = pair<int, int>;
const int inf = 0x3f3f3f3f;
const int maxn = 1e5 + 5;
const int dir[4][2] = { {-1,0},{0,-1},{1,0},{0,1} };
int p[100];
int n,sum;
int a[100],b[100],c[100],d[100];//c为主对角线
void inti(){
memset(b,0,sizeof(b));
memset(c,0,sizeof(c));
memset(d,0,sizeof(d));
sum=0;
}
void dfs(int q){
if(q>n){
sum++;
return;
}
for(int i=1;i<=n;i++){
if(b[i]==0&&c[i-q+n]==0&&d[i+q]==0){
b[i]=1;
c[i-q+n]=1;
d[i+q]=1;
dfs(q+1);
b[i]=0;
c[i-q+n]=0;
d[i+q]=0;
}
}
}
int main() {
//ios;
while(scanf("%d",&n)&&n!=0){
if(p[n]!=0){
cout<<p[n]<<endl;
}
else{
inti();
dfs(1);
p[n]=sum;
printf("%d\n",sum);
}
}
return 0;
}