DFS 典型题之 n 皇后(c++版)

n 皇后

前置知识:

根据数学函数 y = x +b;

当不同的一元函数的斜率相同时,唯一能区分他们的就是截距b

因此,可以由 b = y-x 得到不同的对角线直线函数(我们假设这些直线的斜率都是1)

那么反对角线 y = -x +b 所以b = x+y;

正对角线 b = y-x

可能出现b<0 的情况,所以我们给b加一个数当偏移量c,至于这个数是多少无所谓,只要保证这个数 c+y-x>0即可

其实b是多少不重要,重要的是我们能用b来唯一确定这条对角线,作为一个唯一表示,来映射出这条直线

优化版本:

整体思路:

全排列思想

因为每一行只能放一个皇后,只要当这一行出现过皇后,那就进入下一行

所以我们可以遍历每一行的所有列,只要在这个点放了皇后,那么直接进入下一行

#include

int n;
char q[20][20];
bool f[100],zh[100],fn[100];

void dfs(int u){
    //已经遍历完了所有的行,输出答案
    if(u>n){
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++)std::cout<<q[i][j];
            std::cout<<std::endl;
        }
        std::cout<<std::endl;
        return ;
    }
    
    for(int i=1;i<=n;i++){
        //f里面装的是这一列是否已经用过
        //有这三个判断在,只有满足题意的皇后能走到最后一层
        //不符合题意的提前剪枝了
        if(!f[i]&&!zh[i+u]&&!fn[n+u-i]){
            f[i] = zh[i+u] = fn[n+u-i] = true;
            q[u][i] = 'Q';
            dfs(u+1);//进入下一层
             f[i] = zh[i+u] = fn[n+u-i] = false;
             q[u][i] = '.';
        }
    }

}
int main(){

    std::cin>>n;

    for(int i=1;i<=n;i++)for(int j=1;j<=n;j++)q[i][j] = '.';
    dfs(1);
    return 0;

}

原始暴力版

思路:

暴力遍历每一个格子是否能填皇后,所以有两种情况,填和不填

用数组剪枝一下,填的时候还要注意得能填,不能填的位置不能填,实现剪枝的效果

能填的情况:这一行,这一列,这一正对角线,这一反对角线都无皇后

#include

const int N=100;
int n;
bool row[N],col[N],pl[N],inv[N];//行,列,正对角线,反对角线
char q[N][N];

void dfs(int x,int y,int num){ //num当前已经有多少八皇后存进去了
    if(y>n)y = 1,x++;
    if(x>n){
        if(num==n){
            for(int i=1;i<=n;i++){
                for(int j=1;j<=n;j++)std::cout<<q[i][j];
                std::cout<<std::endl;
            }
            std::cout<<std::endl;
        }
        return ;
    }
    //不填八皇后
    dfs(x,y+1,num);
    if(!row[x]&&!col[y]&&!pl[x+y]&&!inv[n+y-x]){
        row[x] = col[y] = pl[x+y] = inv[n+y-x]  = true;
        q[x][y] = 'Q';
      
        //因为这一行填了皇后,进入下一行就要从第一列开始判断
        //num可以写在传参也可以写在外面
        //例如 num++;  dfs(x+1,1);num--;
        //记得恢复现场就行
        dfs(x+1,1,num+1);
        row[x] = col[y] = pl[x+y] = inv[n+y-x]  = false;
        q[x][y] = '.';
    }
}
int main(){
    std::cin>>n;
    for(int i=1;i<=n;i++)for(int j = 1;j<=n ;j++)q[i][j] = '.';
    dfs(1,1,0);
    return 0;
}

你可能感兴趣的:(Acm算法,c++,深度优先,c++,算法)