HDU 2553 - N皇后问题 [回溯法]

Problem Description

在N*N的方格棋盘放置了N个皇后，使得它们不相互攻击（即任意2个皇后不允许处在同一排，同一列，也不允许处在与棋盘边框成45角的斜线上。你的任务是，对于给定的N，求出有多少种合法的放置方法。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

int N, cnt, vis[3][32], res[12];

void Search(int cur){
	if(cur == N) {cnt++; return;}
	for(int i = 0; i < N; ++i){
		if(!vis[0][i] && !vis[1][cur+i] && !vis[2][cur-i+N]){
			vis[0][i] = vis[1][cur+i] = vis[2][cur-i+N] = 1;
			Search(cur + 1);
			vis[0][i] = vis[1][cur+i] = vis[2][cur-i+N] = 0;
		}
	}
}
int main(int argc, char *argv[]) {
	while (cnt = 0, memset(vis, 0, sizeof(vis)), cin >> N, N) {
        if(!res[N]) Search(0), res[N] = cnt;
		cout << res[N] << endl;
	}
	return 0;
}