【NEEPU OJ】3001--N皇后问题

描述

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

输入

共有若干行，每行一个正整数N≤10，表示棋盘和皇后的数量；如果N=0，表示结束。

输出

共有若干行，每行一个正整数，表示对应输入行的皇后的不同放置数量。

输入样例 1

1
8
5
0

输出样例 1

1
92
10

---

代码

#include <iostream>
using namespace std;

int bit, sum;

void queens(int row, int ld, int rd){
    int pos, p;
    if(bit != row){
        pos = bit & ~(row | ld | rd);
        while(pos){
            p = pos & -pos;
            pos -= p;
            queens(row + p, (ld + p) << 1, (rd + p) >> 1);
        }
    }
    else{
        ++sum;
    }
}

int main(){
    int n;
    while(cin >> n){
        if(n == 0){
            break;
        }
        sum = 0;
        bit = (1 << n) - 1;
        queens(0, 0, 0);
        cout << sum << endl;
    }
    return 0;
}