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