DFS-n皇后问题

n皇后问题

n-皇后问题是指将 n 个皇后放在 n∗n 的国际象棋棋盘上，使得皇后不能相互攻击到，即任意两个皇后都不能处于同一行、同一列或同一斜线上。

思路

第一种思路

搜索全排列：枚举每一行皇后放到哪个位置上去，判断当前方案是否合法，剪枝操作。

code

现在给定整数n，请你输出所有的满足条件的棋子摆法。

输入格式

共一行，包含整数n。

输出格式

每个解决方案占n行，每行输出一个长度为n的字符串，用来表示完整的棋盘状态。

其中”.”表示某一个位置的方格状态为空，”Q”表示某一个位置的方格上摆着皇后。

每个方案输出完成后，输出一个空行。

数据范围

1≤n≤91≤n≤9

输入样例：

4

输出样例：

.Q..
...Q
Q...
..Q.

..Q.
Q...
...Q
.Q..

#include <iostream>

using namespace std;

const int N = 20;//开两倍正反对角线

int n;
char g[N][N];//用字符串来记找到的方案
int path[N];//存状态
bool col[N], dg[N], udg[N];//col同一列只能有一个，正对角线，反对角线都开一个
//当前位置需要填哪些数，需要开bool数组，如果该点为true则表示该点被用过了

void dfs(int u)
{
    //表示已经搜了n行，输出
    if (u == n)
    {
        for (int i = 0; i < n; i ++) puts(g[i]);
        puts("");//换行
        return;
    }
     //对n个位置按行搜索

    for (int i = 0; i < n; i ++ )//枚举当前位置能填哪些数
    //     //确保这一列没放过，正（x + b)对角线没放过反(-x + b)过，+n是给个偏移量
        if (!col[i] && !dg[u + i] && !udg[n - u + i])//如果该数没被用过
        {
            g[u][i] = 'Q';//将i放到当前位置
            col[i] = dg[u + i] = udg[n - u + i] = true;//记录i已被用过
            dfs(u + 1);//递归到下一层
            //当dfs结束之后，需要恢复现场
            col[i] = dg[u + i] = udg[n - u + i] = false;
            g[u][i] = '.';
        }
    
}


int main()
{
    //共一行，包含一个整数n
    cin >> n;
    
    for (int i = 0; i < n; i ++)
        for (int j = 0; j < n; j ++)
            g[i][j] = '.';
    
    dfs(0);//从第0个位置开始
    
    return 0;
}

第二种思路

挨个枚举所有格子

Code

#include <iostream>

using namespace std;

const int N = 20;//开两倍正反对角线

int n;
char g[N][N];//用字符串来记找到的方案

bool row[N], col[N], dg[N], udg[N];//col同一列只能有一个，正对角线，反对角线都开一个
//当前位置需要填哪些数，需要开bool数组，如果该点为true则表示该点被用过了

//挨个枚举所有格子 
void dfs(int x, int y, int s)
{
    //判断y == n，如果出界，变到下一行格子 x ++
    if (y == n) y = 0, x ++;
    
    if (x == n)//如果枚举完最后一行
    {
        if (s == n)//找到一组解
        {
            for (int i = 0; i < n; i ++) puts(g[i]);//输出
            puts("");
        }
        return;
    }
    //不放皇后，直接递归到下一个格子 y + 1
    dfs(x, y + 1, s);
    
    if (!row[x] && !col[y] && !dg[x + y] && !udg[x - y + n])//如果该数没被用过
        {
            //放皇后
            g[x][y] = 'Q';//将i放到当前位置
            row[x] = col[y] = dg[x + y] = udg[x - y + n] = true;//更新记录，都已经放上皇后
            dfs(x, y + 1, s + 1);//递归到下一层
            //当dfs结束之后，需要恢复现场
            row[x] = col[y] = dg[x + y] = udg[x - y + n] = false;//恢复现场
            g[x][y] = '.';
        }
    
}


int main()
{
    //共一行，包含一个整数n
    cin >> n;
    
    for (int i = 0; i < n; i ++)
        for (int j = 0; j < n; j ++)
            g[i][j] = '.';
    
    //输入：x, y, s:左上角开始搜并记录有多少个皇后s
    dfs(0, 0, 0);//从第0个位置开始
    
    return 0;
}