【acwing】843. n-皇后问题

题目描述

解题思路

思路一:O(n!)

• 一行一行来看，从第0行开始，我们依次给每一行放置一个皇后，对于一个确定的行，我们只需要去枚举放置的列即可，在保证放置合法的情况下，一直搜索下去。

• 确定了搜索策略，现在我们还有一个问题没有解决，那么就是如何判断是否发生了冲突。因为我们是逐行放置的，所以行内是不会冲突的，对于列也很好处理，如果某列被占用了，只需要用一个数组标记即可。

• 而最难处理的是同一斜线了。一种方法是我们可以直接暴力枚举这个皇后对应的两条对角线上的点，判断是否有冲突。但是这种方法，既麻烦，效率也比较低。下面我们介绍一种巧妙的方法。
  
  有一个规律，这条对角线上的位置的行加列的值相同，都是 4
  
  而这样的对角线上的位置的坐标的行减去列的值也是相同的，都是-2。

而我们发现，对于每条对角线,和值或者差值都是不一-样的， 而我们正好可以用这一-点来标记一 条对角线是否被占用, 使得标记对角线就像标记列一样简单了。

#include<iostream>
using namespace std;
const int N=20;
char a[N][N];//存储路径
//col列,dg对角线，udg反对角线
int col[N],dg[N],udg[N];
int n;
void dfs(int u){
    if(u==n){/已经到了第n行
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                cout<<a[i][j];
            }
            cout<<endl;
        }
        cout<<endl;
        return ;
    }
    
    for(int i=0;i<n;i++){
        if(!col[i] && !dg[u+i] && !udg[i-u+n]){//这一列，两条对角线没有标记过
            a[u][i]='Q';
            col[i]=dg[u+i]=udg[i-u+n]=1;//标记
            dfs(u+1);
            col[i]=dg[u+i]=udg[i-u+n]=0;//取消标记
            a[u][i]='.';
        }
    }
}

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

思路二：O(2ⁿ²)

依次按格子中的每个位置来枚举

// 不同搜索顺序 时间复杂度不同  所以搜索顺序很重要！
#include <iostream>
using namespace std;
const int N = 20;

int n;
char g[N][N];
//row行，col列，dg对角线，udg反对角线
bool row[N], col[N], dg[N], udg[N]; // 因为是一个个搜索，所以加了row

// s表示已经放上去的皇后个数
void dfs(int x, int y, int s)
{
    // 处理超出边界的情况，这里是当y到边界的时候，x往下一行走
    if (y == n) y = 0, x ++ ;

    if (x == n) { // x==n说明已经枚举完n^2个位置了
        if (s == n) { // s==n说明成功放上去了n个皇后
            for (int i = 0; i < n; i ++ ) puts(g[i]);
            puts("");
        }
        return;
    }

    // 分支1：放皇后
    if (!row[x] && !col[y] && !dg[x + y] && !udg[x - y + n]) {
        g[x][y] = 'Q';
        row[x] = col[y] = dg[x + y] = udg[x - y + n] = true;
        dfs(x, y + 1, s + 1);
        row[x] = col[y] = dg[x + y] = udg[x - y + n] = false;
        g[x][y] = '.';
    }

    // 分支2：不放皇后，向下一个格子移动
    dfs(x, y + 1, s);
}


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);//（x坐标，y坐标，皇后个数）

    return 0;
}

其他大佬总结

dfs解题套路
dfs走迷宫动态演示