题目描述
在8*8
的棋盘上放置8
个皇后,使它们互不攻击,即任意两个皇后不允许处在同一横排。同一纵列,也不允许处在同一与棋盘边框成45°
角的斜线上。现在假设第一行的皇后的位置已经确定了,问符合条件的摆法有多少种?
输入
多组测试数据。先输入一个整数T
表示组数。 然后是T
组数据。每组数据一行,输入一个整数n
( 1 <= n <= 8
),表示第1
行的皇后放的列数
输出
对于每组数据输出一行,值为满足条件摆法的种数。
样例输入 Copy
1
1
样例输出 Copy
4
经典DFS
AC
代码
#pragma GCC optimize(2)
#include<bits/stdc++.h>
#define abss(x) ((x)>(0)?(x):(-1)*(x))
#define maxs(a,b) ((a)>(b)?(a):(b))
#define mins(a,b) ((a)<(b)?(a):(b))
#define FOR(i,a,b) for(register int i=(a);i<=(b);i++)
#define ROF(i,a,b) for(register int i=(a);i>=(b);i--)
#define mem(a) memset(a,0,sizeof(a))
const int INF (1<<30);
const int inf (-1<<30);
using namespace std;
int matx[10][10];
int ans;
bool check_edge(int x,int y){
if(x>=1 and x<=8 and y>=1 and y<=8)
return true;
return false;
}
bool check_unblocked(int x,int y){
FOR(i,1,8){
if(matx[i][y]==1)return false;
if(matx[x][i]==1)return false;
}
int tx=x,ty=y;
while(check_edge(tx,ty)) if(matx[tx++][ty++]==1)return false;
tx=x,ty=y;
while(check_edge(tx,ty)) if(matx[tx--][ty++]==1)return false;
tx=x,ty=y;
while(check_edge(tx,ty)) if(matx[tx++][ty--]==1)return false;
tx=x,ty=y;
while(check_edge(tx,ty)) if(matx[tx--][ty--]==1)return false;
return true;
}
void dfs(int x,int y,int queen){
if(x>queen)return;
if(queen==8){
ans++;
return;
}
FOR(nx,x+1,8)
FOR(ny,1,8){
if(matx[nx][ny]==0 and check_unblocked(nx,ny)){
matx[nx][ny]=1;
dfs(nx,ny,queen+1);
matx[nx][ny]=0;
}
}
}
int main(){
int T;
cin>>T;
while(T--){
int n;
mem(matx);
ans=0;
cin>>n;
matx[1][n]=1;
dfs(1,n,1);
cout<<ans<<endl;
}
return 0;
}