843. n-皇后问题

 DFS经典问题

 1.用全排列的方法：

#include<iostream>

using namespace std;

const int N = 20;

int n;
char g[N][N];
bool col[N],dg[N],udg[N];//对角线

void dfs(int u)
{
    if(u==n)
    {
        for(int i=0;i<n;i++) puts(g[i]);//二维数组的输出
        puts("");
        return;
    }
    
    for(int i=0;i<n;i++)
        if(!col[i]&&!dg[n+u-i]&&!udg[u+i])//用不同截距表示不同对角线，+n是使对角线有偏移量从而区分
        {
            g[u][i]='Q';
            col[i]=dg[n+u-i]=udg[u+i]=true;//标记行、列、对角线
            dfs(u+1);
            col[i]=dg[n+u-i]=udg[u+i]=false;
            g[u][i]='.';
        }
}

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

2.用每个空格都有放与不放两种选择的思路

#include<iostream>

using namespace std;

const int N = 20;

int n;
char g[N][N];
bool row[N],col[N],dg[N],udg[N];//对角线
//从左上角开始挨个搜索
void dfs(int x,int y,int s)
{
    if(y==n) y=0,x++;//
    
    if(x==n)
    {
        if(s==n)
        {
            for(int i=0;i<n;i++) puts(g[i]);
            puts("");
        }
        return;
    }
    
    dfs(x,y+1,s);//不放皇后
    
    if(!row[x]&&!col[y]&&!dg[n+y-x]&&!udg[y+x])//放皇后
    {
        g[x][y]='Q';
        row[x]=col[y]=dg[n+y-x]=udg[y+x]=true;
        dfs(x,y+1,s+1);
        row[x]=col[y]=dg[n+y-x]=udg[y+x]=false;
        g[x][y]='.';
    }
}

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