回溯是搜索算法中的一种控制策略,基本思想是选择一条路向前探索,如果走不通就退回来换另一条路走,如此反复直到得到最终结果,回溯算法中常常用到递归与递推,它把问题的解空间转化成了图或者树的结构表示,然后使用深度优先的方式搜索。
列举一道经典题理解回溯思想
八皇后问题
要在国际象棋棋盘中放8个皇后,使任意两个皇后都不能互相吃。(提示:皇后能吃同一行、同一列、同一对角线的任意棋子)
分析:这里用到了dfs,dfs中就用到了回溯思想
代码
#include<bits/stdc++.h>
#define N 100
using namespace std;
int a[N][N],b[N];
int vis[N][N];
int tot;
int dir[4][2]={{0,1},{0,-1},{1,0},{-1,0}};
void dfs(int step)
{
if(step==8+1)
{
tot++;
for(int i=1;i<=8;i++)
a[tot][i]=b[i];
return;
}
for(int i=1;i<=8;i++)
{
if(vis[0][i]==0&&vis[1][step+i]==0&&vis[2][step-i+8]==0)
{
vis[0][i]=1;
vis[1][i+step]=1;
vis[2][step-i+8]=1;
b[step]=i;
dfs(step+1);
vis[0][i]=0;
vis[1][i+step]=0;
vis[2][step-i+8]=0;
}
}
}
int main()
{
dfs(1);
for(int t=1;t<=tot;t++)
{
cout<<"No."<<t<<endl;
for(int i=1;i<=8;i++)
{
for(int j=1;j<=8;j++)
{
if(a[t][j]==i)
cout<<"1 ";
else
cout<<"0 ";
}
cout<<endl;
}
}
return 0;
}