[BZOJ]4809: 皇后

4809: 皇后

Description

众所不知，rly现在不会玩国际象棋。但是，作为一个OIer，rly当然做过八皇后问题。这里再啰嗦几句，皇后可以攻击到同行同列同对角线，在n*n的方格中摆n个皇后使其互不攻击到，求不同的解的数量，这就是经典的n皇后问题。现在问题推广到n皇后问题，这个问题对于你而言实在是小菜一叠。但因为上一次rly把棋盘弄破了，又拿不出新的，所以rly打算难一点点，问题就是破棋盘上的n皇后问题。他想知道……（你们懂的）。

棋子都是相同的。

Input

一行，一个正整数N。

接下来N行，每行N个数，要么为0，表示没坏，要么1，表示坏了。

N<=16

Output

一行，输出不同的解的数量。

10s的题9992ms卡过。。。裸n皇后问题。

#include<cstdio>
#include<cstring>
const int Q=18;
int n,a[Q][Q],ans=0;
bool r[Q],x1[Q*2],x2[Q*2];
void dfs(int x)
{
    if(x==n+1)
    {
        ans++;
        return;
    }
    for(int i=1;i<=n;i++)
    if(a[x][i]==0 && r[i]==false && x1[x-i+n]==false && x2[x+i]==false)
    {
        r[i]=true;x1[x-i+n]=true;x2[x+i]=true;
        dfs(x+1);
        r[i]=false;x1[x-i+n]=false;x2[x+i]=false;
    }
}
int main()
{
    memset(r,false,sizeof(r));
    memset(x1,false,sizeof(x1));
    memset(x2,false,sizeof(x2));
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++)
    scanf("%d",&a[i][j]);
    dfs(1);
    printf("%d",ans);
}