DFS合集

深度优先搜索算法的通用解法

 

深度优先搜索算法(Depth First Search),是图论中的经典算法。

深度优先搜索算法是沿着树的深度遍历树的节点,尽可能深的搜索树的分支。当结点所有子结点那一层都被搜索过,再回溯返回到当前结点的邻结点,继续搜索,直到遍历完整棵树。一般采用的是前序遍历,先根然后再左右结点的方式进行。

一些经典的问题,比如八皇后、马走日、迷宫等,都可以通过深度优先搜索算法来解决。

为了方便描述,下文用DFS来做为深度优先搜索算法的简称。

二、我对DFS的认识 对于DFS,我相信很多人第一次接触很难设计出相应的算法,即便是有不错的编程经验。我第一次几乎没办法设计出解决八皇后的算法,即便是想了很久。最后没办法只好参照别人写的递归式的DFS。之后,虽然对这个算法有一点了解,但由于了解不够深度,过了几天就记不得了,下次又完全不知道怎么入手。然后需要再到网上搜下代码,看一遍后大概才双知道。而且发现每次写代码的时候心里总觉得不踏实,一开始总有错误的地方,并且每次写的代码都有些不同。总之,写过很多次后,依然是停留到了解的阶段,没办法进一步提升,特别是非递归式的DFS一直都停留到靠脑力记忆而不是理解的阶段。

今天周末有点时间,觉得有必要解决这些问题,试着花时间去归纳总结DFS的本质,看能否做到一劳永逸。

我设定的目标是:

1、不仅停留到理解阶段,而是要知道这个算法每一步的实现

2、捉住其中的本质,给出这个算法的设计框架。

3、在1与2的基础中,可以熟练写出递归与非递归两种实现方式 。

扫描二维码关注公众号,回复: 5694744 查看本文章

经过一个下午的研究,我发现任何DFS只需要通过下面几步就可以实现,无论是递归还是非递归方式。我给这几步分别做了一个命名,分别是find、forward、done、back。

如下:

1、find(right):在树的当前层,横向遍历,尝试找到ok的节点。(这一步通常被叫做剪枝,只留下ok的。)

2、forward(down):若在当前层找到ok的结点,并且当前层不是最后一层:把ok的节点放到当前层;进入下一层第一个结点。跳到find

3、done(right):若在当前层找到ok的结点,并且当前层是最后一层:打印出结果;进入当前层的下一个结点。跳到find

4、back(up):在当前层没有找到ok的节点:返回上一层当前结点的下一个兄弟节点。跳到find

其实最重要的是find。然后后面的forward、done、back只是用来控制搜索走向。这四步可以进一步总结成两步。 为了了解算法,我想最好的切入方式是从一些实例开始。下面分别从八皇后以及马走日等问题做为切入点来分析DFS

三、用DFS解八皇后

1、问题描述

八皇后问题是一个以国际象棋为背景的问题:如何能够在8×8 的国际象棋棋盘上放置八个皇后,使得任何一个皇后都无法直接吃掉其他的皇后?

也就是说,使得棋盘中每个横向、纵向、左上至右下斜向、右上至左下斜向均只有一枚皇后。

八皇后有92组解,下面给出其中一种解的图例:

Image

2、 问题分析

规则是每一个皇后与前面的所有皇后不能在同一行、同一列、同一对角线。我们可以从第0行,第0列开始摆放,然后按照深度优先的原则,按照规则往更下面的行摆放皇后,直到摆放完8行。因为解不只一个,当某一行(包括最后一行跟最后一行之前的所有行)的所有列都被尝试过,再回溯返回到上一行,继续深度优先,直到遍历完整个棋盘的所有情况。得出所有的解。 八皇后问题可以看成是在深度为8的8叉树中,找出所有的解。

3、代码实现

递归算法:

#include <stdio.h>
#include <math.h>

/*八皇后问题是在8*8的棋盘上放置8枚皇后,使得棋盘中每个横向、纵向、左上至右下斜向、右上至左下斜向均只有一枚皇后。
求解出所有摆法,一共有92种摆法*/

const int N = 8; //棋盘行数 int a[N] = {0}; //表示棋盘,若a[2]=2,则表示在第3行第2列放一个皇后,因为同一行不能放两个皇后,所以只需要1维数组就可以表示一个棋盘。 int solution = 0;//解的个数 //row行,col列, 是否可以摆皇后 bool IsOK(int row, int col) {  for (int i = 0; i < row; i++)  {   if (a[i] == col || (abs(a[i] - col) == row - i))   {    return false;   }  }  return true; } void Display() {  printf("第%d种解:\n",++solution);  for (int i = 0; i < N; i++)  {   for (int j = 0; j < N; j++)   {    if (a[i] == j)    {     printf("%d", i);    }    else    {     printf("#");    }   }   printf("\n");  }  printf("-----------------\n"); } void DSF(int row) {  for (int col = 0; col < N; col++)  {   //find   if (IsOK(row, col))   {    a[row] = col;    //forward    if (row != N -1)    {     DSF(row + 1);    }    else    {     //done     Display();    }      }  }  //back } int main() {  DSF(0);  return 0; }

非递归算法:

  1.  
    #include <stdio.h>
  2.  
    #include <math.h>
  3.  
    #include <stack>
  4.  
    using namespace std;
  5.  
    /*八皇后问题是在8*8的棋盘上放置8枚皇后,使得棋盘中每个横向、纵向、左上至右下斜向、右上至左下斜向均只有一枚皇后*/
  6.  
     
  7.  
    const int N = 8; //棋盘行数
  8.  
    int a[N] = {0}; //表示棋盘,若a[2]=2,则表示在第3行第2列放一个皇后,因为同一行不能放两个皇后,所以只需要1维数组就可以表示一个棋盘。
  9.  
     
  10.  
    int solution = 0;//解的个数
  11.  
     
  12.  
    struct Node
  13.  
    {
  14.  
    int row;
  15.  
    int col;
  16.  
    };
  17.  
     
  18.  
    //row行,col列, 是否可以摆皇后
  19.  
    bool IsOK(Node node)
  20.  
    {
  21.  
    for (int i = 0; i < node.row; i++)
  22.  
    {
  23.  
    if (a[i] == node.col || (abs(a[i] - node.col) == node.row - i))
  24.  
    {
  25.  
    return false;
  26.  
    }
  27.  
    }
  28.  
    return true;
  29.  
    }
  30.  
     
  31.  
    //打印出所有解
  32.  
    void Print()
  33.  
    {
  34.  
    printf("第%d种解:\n", ++solution);
  35.  
    for (int i = 0; i < N; i++)
  36.  
    {
  37.  
    for (int j = 0; j < N; j++)
  38.  
    {
  39.  
    if (a[i] == j)
  40.  
    {
  41.  
    printf("%d", i);
  42.  
    }
  43.  
    else
  44.  
    {
  45.  
    printf("#");
  46.  
    }
  47.  
    }
  48.  
    printf("\n");
  49.  
    }
  50.  
     
  51.  
    printf("-----------------\n");
  52.  
    }
  53.  
     
  54.  
    void DSF()
  55.  
    {
  56.  
    Node node;
  57.  
    stack
  58.  
     
  59.  
     
  60.  
     
  61.  
    stack;
  62.  
     
  63.  
    node.row = 0;
  64.  
    node.col = 0;
  65.  
    stack.push(node);
  66.  
    while(stack.size() >= 1)
  67.  
    {
  68.  
    //--find
  69.  
    node = stack.top();
  70.  
    while (node.col < N && !IsOK(node))
  71.  
    {
  72.  
    node.col++;
  73.  
    }
  74.  
     
  75.  
    if (node.col < N)
  76.  
    {
  77.  
    //--forward
  78.  
    if (node.row < N-1)
  79.  
    {
  80.  
    //把ok的节点放到当前层
  81.  
    a[node.row] = node.col;
  82.  
    stack.pop();
  83.  
    stack.push(node);
  84.  
     
  85.  
    //进入下一层的第一个节点
  86.  
    node.row++;
  87.  
    node.col = 0;
  88.  
    stack.push(node);
  89.  
    }
  90.  
    else
  91.  
    {
  92.  
    //--done
  93.  
    a[node.row] = node.col;
  94.  
    Print();
  95.  
     
  96.  
    //进入当前层的下一个结点
  97.  
    //node = stack.top();
  98.  
    node.col++;
  99.  
    stack.pop();
  100.  
    stack.push(node);
  101.  
    }
  102.  
    }
  103.  
    else
  104.  
    {
  105.  
    //--back
  106.  
    stack.pop();
  107.  
     
  108.  
    if (stack.size() == 0)
  109.  
    {
  110.  
    return;
  111.  
    }
  112.  
    node = stack.top();
  113.  
    node.col++;
  114.  
    stack.pop();
  115.  
    stack.push(node);
  116.  
     
  117.  
    }
  118.  
     
  119.  
    }
  120.  
     
  121.  
    }
  122.  
     
  123.  
    int main()
  124.  
    {
  125.  
    DSF();
  126.  
    return 0;
  127.  
    }
  128.  
     
  129.  
     
  130.  
     
  131.  
     
  132.  
     

三、马走日

1、问题描述

在n*n的棋盘中,马只能走"日"字。马从位置(0,0)出发,把棋盘的每一格都走一次且只走一次。找出所有路径。 5*5的棋盘上,有304种解。

下面是其中一种路径的图例:

Image

2、问题分析

搜索过程是从(0,0)出发,按照深度优先的原则,从8个方向中尝试一个可以走的点,直到尝试过所有的方向,走完棋盘上的所有点,得出所有的解。

马走日问题可以看成是在层数为n*n的8叉树中,找出所有的解。

3、代码实现

同样的,也可以把上面的算法框架,套用于马走日的身上。

递归算法:

#include <stdio.h>

/*马走日*/

const int N = 5; //棋盘行数跟列数 int matrix[N][N] = {0}; //表示棋盘 int solution = 0;//解的个数 int count = 0; //第几步 int move[8][2]={{-1,-2},{-2,-1}, {-2,1},{-1,2},{1,2},{2,1},{2,-1},{1,-2}};//八个方向 //在棋盘范围内,而且可放棋 bool IsOK(int x, int y) {  if(( x <= N-1 ) && (x >=0 )   && (y <= N-1 ) && (y >=0 )   && (matrix[x ][y ]==0 ))  {   return true;  }  else  {   return false;  } } //打印出所有解 void Display() {  printf("第%d种解:\n",++solution);  for (int i = 0; i < N; i++)  {   for (int j = 0; j < N; j++)   {       printf("%3d",matrix[i][j]);   }   printf("\n");  }   printf("-----------------\n"); } void DFS(int x, int y) {  int nextX, nextY;  for (int i = 0; i < 8; i++)  {   nextX = x + move[i][0];   nextY = y + move[i][1];   //--find   if (IsOK(nextX, nextY))   {       if (count != N*N -1 )    {     //--forward     count++;     matrix[nextX][nextY] = count;         DFS(nextX, nextY);     matrix[nextX][nextY] = 0;     count--;    }    else    {     //--done     Display();    }   }  }  //--back } int main() {  matrix[0][0] = 1;  count = 1;  DFS(0, 0);  return 0; }

非递归算法:

  1.  
    #include <stdio.h>
  2.  
    #include <stack>
  3.  
    using namespace std;
  4.  
     
  5.  
    /*马走日*/
  6.  
     
  7.  
    const int N = 5; //棋盘行数跟列数
  8.  
    int matrix[N][N] = {0}; //表示棋盘
  9.  
    int solution = 0;//解的个数
  10.  
    int count = 0; //第几步
  11.  
    int move[8][2]={{-1,-2},{-2,-1}, {-2,1},{-1,2},{1,2},{2,1},{2,-1},{1,-2}};//八个方向
  12.  
     
  13.  
    //注意find这一步当前层的的结点,结点的坐标不是x与y,而通过Node中的x与y与direction三者计算后得到当前层的结点
  14.  
    struct Node
  15.  
    {
  16.  
    int x;
  17.  
    int y;
  18.  
    int direction;
  19.  
    };
  20.  
     
  21.  
    //在棋盘范围内,而且可放棋
  22.  
    bool IsOk(Node node)
  23.  
    {
  24.  
    int x, y;
  25.  
    x = node.x + move[node.direction][0];
  26.  
    y = node.y + move[node.direction][1];
  27.  
    if(( x <= N-1 ) && (x >=0 )
  28.  
    && (y <= N-1 ) && (y >=0 )
  29.  
    && (matrix[x][y]==0 ))
  30.  
    {
  31.  
    return true;
  32.  
    }
  33.  
    else
  34.  
    {
  35.  
    return false;
  36.  
    }
  37.  
    }
  38.  
     
  39.  
    //打印
  40.  
    void Print()
  41.  
    {
  42.  
    printf("第%d种解:\n",++solution);
  43.  
    for (int i = 0; i < N; i++)
  44.  
    {
  45.  
    for (int j = 0; j < N; j++)
  46.  
    {
  47.  
     
  48.  
    printf("%3d",matrix[i][j]);
  49.  
    }
  50.  
    printf("\n");
  51.  
    }
  52.  
     
  53.  
    printf("-----------------\n");
  54.  
    }
  55.  
     
  56.  
     
  57.  
    void DFS()
  58.  
    {
  59.  
    Node node;
  60.  
    stack
  61.  
     
  62.  
     
  63.  
     
  64.  
    stack;
  65.  
    int x, y;
  66.  
    count = 1;
  67.  
    node.x = 0;
  68.  
    node.y = 0;
  69.  
    node.direction = 0;
  70.  
    matrix[0][0] = count++;
  71.  
    stack.push(node);
  72.  
     
  73.  
    node.direction = 0;
  74.  
    stack.push(node);
  75.  
     
  76.  
    while(stack.size() >= 2)
  77.  
    {
  78.  
    //--find
  79.  
    node = stack.top();
  80.  
    while (node.direction < 8 && !IsOk(node))
  81.  
    {
  82.  
    node.direction++;
  83.  
    }
  84.  
    if (node.direction < 8)
  85.  
    {
  86.  
    //--forward
  87.  
    if (count < N * N)
  88.  
    {
  89.  
    //把ok的节点放到当前层
  90.  
    stack.pop();
  91.  
    stack.push(node);
  92.  
     
  93.  
    x = node.x + move[node.direction][0];
  94.  
    y = node.y + move[node.direction][1];
  95.  
    matrix[x][y] = count++;
  96.  
     
  97.  
    //进入下一层的第一个节点
  98.  
    node.x = x;
  99.  
    node.y = y;
  100.  
    node.direction = 0;
  101.  
    stack.push(node);
  102.  
    }
  103.  
    else
  104.  
    {
  105.  
    //--done
  106.  
    //打印出结果;
  107.  
    x = node.x + move[node.direction][0];
  108.  
    y = node.y + move[node.direction][1];
  109.  
    matrix[x][y] = count++;
  110.  
    Print();
  111.  
    //注意先清除当前结点的数据
  112.  
    matrix[x][y] = 0;
  113.  
    count--;
  114.  
    //进入当前层的下一个结点;
  115.  
    node.direction++;
  116.  
    stack.pop();
  117.  
    stack.push(node);
  118.  
    }
  119.  
    }
  120.  
    else
  121.  
    {
  122.  
     
  123.  
    //----back
  124.  
    //返回上一层当前结点的下一个节点
  125.  
    stack.pop();
  126.  
    if (stack.size() == 1)
  127.  
    {
  128.  
    return;
  129.  
    }
  130.  
    node = stack.top();
  131.  
    //注意先清除当前结点的数据
  132.  
    x = node.x + move[node.direction][0];
  133.  
    y = node.y + move[node.direction][1];
  134.  
    matrix[x][y] = 0;
  135.  
    count--;
  136.  
     
  137.  
    node.direction++;
  138.  
    stack.pop();
  139.  
    stack.push(node);
  140.  
     
  141.  
    }
  142.  
    }
  143.  
    }
  144.  
     
  145.  
     
  146.  
    int main()
  147.  
    {
  148.  
    DFS();
  149.  
    return 0;
  150.  
    }
  151.  
     
  152.  
     
  153.  
     
  154.  
     

四、DFS有更多的变种,但都可以通过上面所说的四个步骤云解决。未完,待续。。。。

猜你喜欢

转载自www.cnblogs.com/946whn/p/10615527.html
dfs