843.n-皇后问题 做题笔记

深度优先遍历DFS

每一行必定有一个皇后，对行进行深度遍历。

对于第 r 行的第 i 个位置，判断每个点是否可以放皇后，如果可以，则放皇后，然后处理 r + 1 行。

直到 r = n，程序执行完毕。

1.用col数组表示同一列有无皇后，0表示无，1表示有。 

2.用dg数组表示同一主斜线有无皇后。主斜线上采样的多个点的x、y值的差，即说明多个点可通过减法映射到数组中的同一位置，判断斜线上是否出现过一次皇后，即查看 dg[x - y] 是否为1即可。但这样做减法时可能会得到一个负数，是没法直接映射到数组中的，因此主斜线上做判断时统一加上一个常量保证非负，即dg[x - y + n]。 如下图：

3.用udg数组表示副斜线上有无皇后。副斜线上采样的多个点的x、y值的和，即说明多个点可通过加法映射到数组中的同一位置，判断斜线上是否出现过一次皇后，即判断 udg[x + y] 是否为1即可。 

注意：dg和udg数组要开两倍的N。（极限） 

#include<iostream>
using namespace std;
const int N=20;
char q[N][N];
int col[N],dg[N*2],udg[N*2],n;

void dfs(int u){
    if(u==n){
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                cout<<q[i][j];
            }
            printf("\n");
        }
        printf("\n");
        return;
    }
    for(int i=0;i<n;i++){
        if(!col[i]&&!dg[n-u+i]&&!udg[i+u]){
            q[u][i]='Q';
            col[i]=dg[n-u+i]=udg[i+u]=1;
            dfs(u+1);
            col[i]=dg[n-u+i]=udg[i+u]=0;
            q[u][i]='.';
        }
    }
}

int main(){
    cin>>n;
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            q[i][j]='.';
        }
    }
    dfs(0);
}