八(n)皇后问题的方法总结

第一种方法是我做洛谷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

猜你喜欢

转载自blog.csdn.net/weixin_43537190/article/details/84728261
今日推荐