BZOJ 4809: 皇后 （n皇后问题）dfs搜索入门

4809: 皇后

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 166  Solved: 89
[Submit][Status][Discuss]

Description

众所不知，rly现在不会玩国际象棋。但是，作为一个OIer，rly当然做过八皇后问题。这里再啰嗦几句，皇后可以攻击到同行同列同对角线，在n*n的方格中摆n个皇后使其互不攻击到，求不同的解的数量，这就是经典的n皇后问题。现在问题推广到n皇后问题，这个问题对于你而言实在是小菜一叠。但因为上一次rly把棋盘弄破了，又拿不出新的，所以rly打算难一点点，问题就是破棋盘上的n皇后问题。他想知道……（你们懂的）。

棋子都是相同的。

Input

一行，一个正整数N。

接下来N行，每行N个数，要么为0，表示没坏，要么1，表示坏了。

N<=16

Output

一行，输出不同的解的数量。

Sample Input

4
1 0 1 1
1 1 1 0
0 1 1 1
1 1 0 1

Sample Output

1

经典的n皇后问题

的变形

对于行列的处理很简单

对于两个方向的对角线

我们把它们看成平面直角坐标系下的直线

可知

对于y=x+k   y-x为定值

对于y=-x+k 同理

所以没什么可说的了

我才不会告诉你我在水经验QwQ

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<queue>
#include<set>
#include<map>
using namespace std;
inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return f*x;
}
const int N=20;
int n,ans;
bool mp[N][N],column[N],diagonal1[N<<1],diagonal2[N<<1];
void dfs(int step)
{
	if(step==n+1)
	{ans++;return ;}
	for(int i=1;i<=n;i++)
	if(!mp[step][i]&&!column[i]&&!diagonal1[step+i]&&!diagonal2[i-step+n])
	{
		column[i]=1;diagonal1[step+i]=1;diagonal2[i-step+n]=1;
		dfs(step+1);
		column[i]=0;diagonal1[step+i]=0;diagonal2[i-step+n]=0;
	}
}
int main()
{
	n=read();
	for(int i=1;i<=n;i++)for(int j=1;j<=n;j++)mp[i][j]=read();
	dfs(1);
	printf("%d\n",ans);
	return 0;
}