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

原文鏈接:http://helloitworks.com/678.html?utm_source=tuicool&utm_medium=referral
一、深度優先搜索
深度優先搜索算法(Depth First Search),是圖論中的經典算法。深度優先搜索算法是沿著樹的深度遍歷樹的節點,盡可能深的搜索樹的分支。當結點所有子結點那一層都被搜索過,再回溯返回到當前結點的鄰結點,繼續搜索,直到遍歷完整棵樹。一般采用的是前序遍歷,先根然后再左右結點的方式進行。一些經典的問題,比如八皇后、馬走日、迷宮等,都可以通過深度優先搜索算法來解決。為了方便描述,下文用DFS來做為深度優先搜索算法的簡稱。
二、我對DFS的認識對于DFS,我相信很多人第一次接觸很難設計出相應的算法,即便是有不錯的編程經驗。我第一次幾乎沒辦法設計出解決八皇后的算法,即便是想了很久。最后沒辦法只好參照別人寫的遞歸式的DFS。之后,雖然對這個算法有一點了解,但由于了解不夠深度,過了幾天就記不得了,下次又完全不知道怎么入手。然后需要再到網上搜下代碼,看一遍后大概才雙知道。而且發現每次寫代碼的時候心里總覺得不踏實,一開始總有錯誤的地方,并且每次寫的代碼都有些不同??傊瑢戇^很多次后,依然是停留到了解的階段,沒辦法進一步提升,特別是非遞歸式的DFS一直都停留到靠腦力記憶而不是理解的階段。
今天周末有點時間,覺得有必要解決這些問題,試著花時間去歸納總結DFS的本質,看能否做到一勞永逸。我設定的目標是:1、不僅停留到理解階段,而是要知道這個算法每一步的實現2、捉住其中的本質,給出這個算法的設計框架。3、在1與2的基礎中,可以熟練寫出遞歸與非遞歸兩種實現方式 。
經過一個下午的研究,我發現任何DFS只需要通過下面幾步就可以實現,無論是遞歸還是非遞歸方式。我給這幾步分別做了一個命名,分別是find、forward、done、back。如下:1、find(right):在樹的當前層,橫向遍歷,嘗試找到ok的節點。(這一步通常被叫做剪枝,只留下ok的。)2、forward(down):若在當前層找到ok的結點,并且當前層不是最后一層:把ok的節點放到當前層;進入下一層第一個結點。跳到find3、done(right):若在當前層找到ok的結點,并且當前層是最后一層:打印出結果;進入當前層的下一個結點。跳到find4、back(up):在當前層沒有找到ok的節點:返回上一層當前結點的下一個兄弟節點。跳到find
其實最重要的是find。然后后面的forward、done、back只是用來控制搜索走向。這四步可以進一步總結成兩步。為了了解算法,我想最好的切入方式是從一些實例開始。下面分別從八皇后以及馬走日等問題做為切入點來分析DFS
三、用DFS解八皇后
1、問題描述八皇后問題是一個以國際象棋為背景的問題:如何能夠在8×8 的國際象棋棋盤上放置八個皇后,使得任何一個皇后都無法直接吃掉其他的皇后?也就是說,使得棋盤中每個橫向、縱向、左上至右下斜向、右上至左下斜向均只有一枚皇后。八皇后有92組解,下面給出其中一種解的圖例:


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

非遞歸算法:

#include <stdio.h>
#include <math.h>
#include <stack>
using namespace std;
/*八皇后問題是在8*8的棋盤上放置8枚皇后,使得棋盤中每個橫向、縱向、左上至右下斜向、右上至左下斜向均只有一枚皇后*/
 
const int N = 8; //棋盤行數
int a[N] = {0}; //表示棋盤,若a[2]=2,則表示在第3行第2列放一個皇后,因為同一行不能放兩個皇后,所以只需要1維數組就可以表示一個棋盤。
 
int solution = 0;//解的個數
 
struct Node
{
 int row;
 int col;
};
 
//row行,col列, 是否可以擺皇后
bool IsOK(Node node)
{
 for (int i = 0; i < node.row; i++)
 {
 if (a[i] == node.col || (abs(a[i] - node.col) == node.row - i))
 {
 return false;
 }
 }
 return true;
}
 
//打印出所有解
void Print()
{
 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()
{
 Node node;
 stack stack;

 node.row = 0;

 node.col = 0;

 stack.push(node);

 while(stack.size() >= 1)

 {

 //--find

 node = stack.top();

 while (node.col < N && !IsOK(node))

 {

 node.col++;

 }

 if (node.col < N)

 {

 //--forward

 if (node.row < N-1)

 {

 //把ok的節點放到當前層

 a[node.row] = node.col;

 stack.pop();

 stack.push(node);

 

 //進入下一層的第一個節點

 node.row++;

 node.col = 0;

 stack.push(node);

 }

 else

 {

 //--done

 a[node.row] = node.col;

 Print();

 

 //進入當前層的下一個結點

 //node = stack.top();

 node.col++;

 stack.pop();

 stack.push(node);

 }

 }

 else

 {

 //--back

 stack.pop();

 

 if (stack.size() == 0)

 {

 return;

 }

 node = stack.top();

 node.col++;

 stack.pop();

 stack.push(node);

 }

 }

}

int main()

{

 DSF();

 return 0;

}

三、馬走日
1、問題描述在nn的棋盤中,馬只能走"日"字。馬從位置(0,0)出發,把棋盤的每一格都走一次且只走一次。找出所有路徑。 55的棋盤上,有304種解。下面是其中一種路徑的圖例:


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

非遞歸算法:
#include <stdio.h>
#include <stack>
using namespace std;
 
/*馬走日*/
 
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}};//八個方向
 
//注意find這一步當前層的的結點,結點的坐標不是x與y,而通過Node中的x與y與direction三者計算后得到當前層的結點
struct Node
{
 int x;
 int y;
 int direction;
};
 
//在棋盤范圍內,而且可放棋
bool IsOk(Node node)
{
 int x, y;
 x = node.x + move[node.direction][0];
 y = node.y + move[node.direction][1];
 if(( x <= N-1 ) && (x >=0 )
 && (y <= N-1 ) && (y >=0 )
 && (matrix[x][y]==0 ))
 {
 return true;
 }
 else
 {
 return false;
 }
}
 
//打印
void Print()
{
 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()
{
 Node node;
 stack stack;

 int x, y;

 count = 1;

 node.x = 0;

 node.y = 0;

 node.direction = 0;

 matrix[0][0] = count++;

 stack.push(node);

 node.direction = 0;

 stack.push(node);

 

 while(stack.size() >= 2)

 {

 //--find

 node = stack.top();

 while (node.direction < 8 && !IsOk(node))

 {

 node.direction++;

 }

 if (node.direction < 8)

 {

 //--forward

 if (count < N * N)

 {

 //把ok的節點放到當前層

 stack.pop();

 stack.push(node);

 x = node.x + move[node.direction][0];

 y = node.y + move[node.direction][1];

 matrix[x][y] = count++;

 

 //進入下一層的第一個節點

 node.x = x;

 node.y = y;

 node.direction = 0;

 stack.push(node);

 }

 else

 {

 //--done

 //打印出結果;

 x = node.x + move[node.direction][0];

 y = node.y + move[node.direction][1];

 matrix[x][y] = count++;

 Print();

 //注意先清除當前結點的數據

 matrix[x][y] = 0;

 count--;

 //進入當前層的下一個結點;

 node.direction++;

 stack.pop();

 stack.push(node);

 }

 }

 else

 {

 //----back

 //返回上一層當前結點的下一個節點

 stack.pop();

 if (stack.size() == 1)

 {

 return;

 }

 node = stack.top();

 //注意先清除當前結點的數據

 x = node.x + move[node.direction][0];

 y = node.y + move[node.direction][1];

 matrix[x][y] = 0;

 count--;

 node.direction++;

 stack.pop();

 stack.push(node);

 

 }

 }

}

int main()

{

 DFS();

 return 0;

}

四、DFS有更多的變種,但都可以通過上面所說的四個步驟云解決。未完,待續。。。。
五、代碼:https://github.com/helloitworks/algorithm/tree/master/dfs

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 229,619評論 6 539
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 99,155評論 3 425
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 177,635評論 0 382
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,539評論 1 316
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 72,255評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,646評論 1 326
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,655評論 3 444
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,838評論 0 289
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 49,399評論 1 335
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 41,146評論 3 356
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 43,338評論 1 372
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,893評論 5 363
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,565評論 3 348
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,983評論 0 28
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 36,257評論 1 292
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 52,059評論 3 397
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 48,296評論 2 376

推薦閱讀更多精彩內容