[USACO1.5] 八皇后 Checker Challenge

提前了解：深度优先搜索(Depth-First-Search)简称DFS

题目描述

一个如下的 n × n 的跳棋棋盘，有六个棋子被放置在棋盘上，使得每行、每列有且只有一个，每条对角线（包括两条主对角线的所有平行线）上至多有一个棋子。n不小于13，求方案数。

分析

本题是经典的搜索回溯例题，考虑到每行只能放一个，且放在第几列都是不同的，可以对n进行枚举全排列然后判断会不会有两个皇后互相攻击。但这样效率太低，可以使用搜索回溯优化。】

#include<bits/stdc++.h>
using namespace std;
int a[15],b[15],c[30],d[30],n,ans;
void dfs(int k)    //第k歌皇后放置
{
	if(k>n)    //如果已经放满
	{
		ans++;    //输出所有答案
		if(ans<=3)//输出前三种答案
		{
			for(int i=1;i<=n;i++)
				cout<<a[i]<<" ";
			cout<<endl;
		}
		return;
	}
	else{
		for(int i=1;i<=n;i++){
			if(b[i]==0&&c[i+k]==0&&d[k-i+15]==0){
				a[k]=i;    //记录放置位置
				b[i]=c[i+k]=d[k-i+15]=1;    //站位
				dfs(k+1);
				b[i]=c[i+k]=d[k-i+15]=0;//取消站位
			}
			
		}
	}
		
}
int main()
{
	cin>>n;//输入
	dfs(1); //调用函数
	cout<<ans;//输出
	return 0;//好习惯
}

纯代码

#include<bits/stdc++.h>
using namespace std;
int a[15],b[15],c[30],d[30],n,ans;
void dfs(int k)
{
	if(k>n)
	{
		ans++;
		if(ans<=3)
		{
			for(int i=1;i<=n;i++)
				cout<<a[i]<<" ";
			cout<<endl;
		}
		return;
	}
	else{
		for(int i=1;i<=n;i++){
			if(b[i]==0&&c[i+k]==0&&d[k-i+15]==0){
				a[k]=i;
				b[i]=c[i+k]=d[k-i+15]=1;
				dfs(k+1);
				b[i]=c[i+k]=d[k-i+15]=0;
			}
			
		}
	}
		
}
int main()
{
	cin>>n;
	dfs(1); 
	cout<<ans;
	return 0;	
}

 最后一道题