DFS深度优先搜索

DFS深度优先搜索

排列数字

题目：

给定一个整数 n，将数字 1∼n 排成一排，将会有很多种排列方法。

现在，请你按照字典序将所有的排列方法输出。

842. 排列数字 - AcWing题库

思路：

根据dfs深度搜索的思路，要对一条路走到黑，好比一棵树，深度搜索类似于树的遍历，是先找到第一个叶子然后回溯，同时对已经找到的叶子进行标记，再去找第二个叶子然后回溯，直到遍历完位置，同时，为了保证该次深度搜索不对下一次深度搜索产生影响，我们需要对该次深度搜索所造成的影响进行恢复（具体看代码实现）。

代码实现：

#include<iostream>

using namespace std;

const int N = 101;

int path[N];
int st[N];
int n;

/*全排列*/
void dfs(int u){
    if(u == n){ //表示此次排列可行，进行输出
        for(int i = 0; i < n; i ++ ) cout << path[i] << " ";
        cout << endl;
        return;
    }
    
    for(int i = 1; i <= n; i ++ ){
        if(st[i] == false) {
            path[u] = i;
            st[i] = true;//表示该数字i被使用
            dfs(u + 1);
            st[i] = false;//恢复，便于下一步深度搜索
        }
    }
}

int main(){
    cin >> n;
    dfs(0);
    return 0;
}

n-皇后问题

题目：

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

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

843. n-皇后问题 - AcWing题库

思路：

类似于全排列的思路，我们对每一行和每一列进行试探，如果可以摆上皇后则继续进行下一步试探，如果失败则此路不可行（也就没有必要进行下一步试探），当我们每一行都摆上了一个皇后之后，则表示此次试探可行，如果没有每一列都摆上，则此次试探失败，于上题类似，深度搜索就是将一条路走到无法再走为止再进行下一位的搜索，直到搜索结束（代码实现对思路进行补充）。

代码实现：

#include<iostream>

using namespace std;

const int N = 20;

int n;
char g[N][N];
bool col[N], dg[N], udg[N]; // 对列，反对角线，对角线进行定义bool数组,进行判断

void dfs(int u){
    if(u == n) { // 表示每一列都找到了皇后
        for(int i = 0; i < n; i ++ ) puts(g[i]);
        puts("");
        return ;
    }
    /*for循环的作用就是重复dfs操作*/
    for(int i = 0; i < n; i ++ ){
        if(!col[i] && !dg[u + i] && !udg[n - u + i]){
        
            /*对此次dfs进行记录*/
            g[u][i] = 'Q';
            col[i] = dg[u + i] = udg[n - u + i] = true;
            
            /*进行下一步试探*/
            dfs(u + 1);
            
            /*对该操作进行恢复，便于下一次dfs*/
            g[u][i] = '.';
            col[i] = dg[u + i] = udg[n - u + i] = false;
        
        }
    }
}

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;
    
}

如有疑问，欢迎评论留言，有问必答，如果对你有帮助，请点赞收藏，mua~