2n皇后问题

优先放置其中一种皇后。

  vis[][]0 1 2 表示黑皇后放置情况

  vis[][]4 5 6 表示白皇后放置情况

放置完黑皇后之后放置白皇后

/*

    Ac码

    在该题的解题过程中，应该注意对观察数组vis[][]第二维的大小 ，二维大小为2n-1 

*/

# include<cstdio>

# include<cstring>

const int maxn = 8;

int p[maxn][maxn];

int n, tot=0;

int vis[maxn][2*maxn];

void dfs(int wq, int bq) {

    

    if (bq == n && wq < n) {//当黑皇后放置完毕时：bq=n，且白皇后放置个数小于n

        for (int i = 0; i < n; i++) {

            if (!vis[4][i] && !vis[5][wq + i] && !vis[6][wq - i + n] && p[wq][i]) {//p[wq][i]表示该位置是否可放置

                vis[4][i] = vis[5][wq + i] = vis[6][wq - i + n] = 1;

                dfs(wq + 1, bq);

                vis[4][i] = vis[5][wq + i] = vis[6][wq - i + n] = 0;

            }

        }

    }

    if (wq == n && bq == n) {

        tot++;

        return;

    }

    else for (int i = 0; i < n; i++) {//放置黑皇后bq

        if (!vis[0][i] && !vis[1][bq + i] && !vis[2][bq - i + n] && p[bq][i]) {

            vis[0][i] = vis[1][bq + i] = vis[2][bq - i + n] = 1;//当前位置放置黑皇后

            p[bq][i] = 0;//标记该位置不可放置

            dfs(wq, bq + 1);//下一行递归

            p[bq][i] = 1;//回标该位置可放置

            vis[0][i] = vis[1][bq + i] = vis[2][bq - i + n] = 0;

            }

    }

}

int main() {

    scanf("%d", &n);

    //input matrix

    memset(vis, 0, sizeof(vis));

    for (int i = 0; i < n; i++) {

        for (int j = 0; j < n; j++) {

            scanf("%d", &p[i][j]);

        }

    }

    dfs(0, 0);

    printf("%d\n", tot);

}