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

来源:互联网 发布:silk labo 知乎 编辑:程序博客网 时间:2024/04/29 17:43
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;}


0 0
原创粉丝点击