八皇后 dfs 洛谷P1219

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;
}