第一种方法是我做洛谷p1219的时候学习的方法 将选的数的八个方向全部做标记,然后如果有标记的话就不访问,深度优化搜索以后再删除标记:
#include<bits/stdc++.h>
using namespace std;
int n,a[50],b[50],c[50],d[50];
int cnt=0;
int print()
{
if(cnt<=3)
{
for(int i=1;i<=n;i++)
{
cout<<a[i]<<" ";
}
cout<<"\n";
}
cnt++;
}
void dfs(int i)
{
if(i>n)
{
print();
return;
}
else
{
for(int j=1;j<=n;j++)
{
if(!b[j]&&(!c[i+j])&&(!d[i-j+n]))
{
a[i]=j;
b[j]=1;
c[i+j]=1;
d[i-j+n]=1;
dfs(i+1);
b[j]=0;
c[i+j]=0;
d[i-j+n]=0;
}
}
}
}
int main()
{
cin>>n;
memset(a,0,sizeof(a));
memset(b,0,sizeof(b));
memset(c,0,sizeof(c));
memset(d,0,sizeof(d));
dfs(1);
cout<<cnt;
return 0;
}
后来发现大佬写的更加简洁,只需要开一个二维数组就可以: 原理一样:
#include <bits/stdc++.h>
using namespace std;
int tot;
int bk[3][1000];
int n;
int num[1000];
void dfs(int cur)
{
if(cur==n)
{
if(tot<3)
{
for(int i=0;i<n;i++)
cout<<num[i]+1<<" ";
cout<<endl;
}
tot++;
return;
}
else for(int i=0;i<n;i++)
{
if(!bk[0][i]&&!bk[1][cur+i]&&!bk[2][cur-i+8])
{
num[cur]=i;
bk[0][i]=bk[1][cur+i]=bk[2][cur-i+8]=1;
dfs(cur+1);
bk[0][i]=bk[1][cur+i]=bk[2][cur-i+8]=0;
}
}
}
main()
{
cin>>n;
dfs(0);
cout<<tot;
} //这个我加过优化了
还有一种就是lrj的写法 这个是我刚开始接触dfs的时候看到的 感觉对我很不友好 = = :
#include<bits/stdc++.h>
using namespace std;
int n=8,c[8];
int tot=0;
void search(int cur)
{
if(cur==n)
tot++;
else
{
for(int i=0;i<n;i++)
{
int ok=1;
c[cur]=i;
for(int j=0;j<cur;j++)
if(c[cur]==c[j]||cur-c[cur]==j-c[j]||cur+c[cur]==j+c[j])
{
ok=0;
break;
}
if(ok)
search(cur+1);
}
}
}
int main()
{
search(0);
cout<<tot;
}
基本就是这几种方法 这是我学习dfs的开始 希望我可以快点理解dfs的内涵把2333