P1219 [USACO1.5]八皇后 Checker Challenge
题意:
n×n的棋盘,放置n个皇后,使其不会互相攻击。同行、同列、同对角线会相互攻击。按字典序输出前三个,第四行输出一共有多少种放法。
做法:
数组a[i]存放第i行的皇后放在哪一列,递归枚举全排列。枚举过程中判合法(可行性剪枝),判断左上、正上、右上是否有冲突,若是,回溯到父亲节点。
代码:
//手动开O2优化
#pragma GCC optimize(2)
#include<iostream>
#include<cstdio>
using namespace std;
int cnt;//计数
bool check(int a[],int n,int cur,int pos){
int temp1=pos,temp2=pos;
for(int j=cur-1;j>=0;j--){
/*检查左上*/ if(temp1>=0 && a[j]==--temp1) return false;
/*检查右上*/ if(temp2<n && a[j]==++temp2) return false;
/*检查正上*/ if(a[j]==pos) return false;
}
return true;
}
//1-n的全排列 存放在a[0:n-1]
void permutation(int a[],int cur,int n){
if(cur==n){
cnt++;
if(cnt<=3) for(int i=0;i<n;i++) printf(i!=n-1 ? "%d ":"%d\n",a[i]);
}
else
for(int i=1;i<=n;i++)
{//尝试在a[cur]中放任意数 i
bool ok=true;
if(!check(a,n,cur,i)) ok=false;
if(ok) a[cur]=i, permutation(a,cur+1,n);
}
}
int main()
{
int n; scanf("%d",&n);
int a[20];
permutation(a,0,n);
printf("%d\n",cnt);
return 0;
}