[演算法] [C++ / Python] 當 DFS 遇上排列
之前說過,DFS 是樹的一種走訪方式,而我們也可以將他應用在「排列」上。
輸出:
根據這樣的想法,我們可以發現 DFS 可以很容易的解決這個問題:將一位數字交由一層 DFS 來處理,而在每一層 DFS 中,我們要做以下這些事情:
1) 檢查是不是最後一層,如果是的話,代表已經得到一組排列,應將結果輸出,並回到上一層
2) 選定一個數字(代表欲排列物之編號),進行下一層的 DFS
讓我們看看 2) 這項敘述的詳細內容:首先是選一個數字,由於我們想讓每樣物品都輪過一次,所以我們使用 for 迴圈來依序取得編號。 接下來很重要的一點,因為有些東西我們可能已經排在前面了,所以我們要用一個陣列 visited[] 來儲存是否用過,如果已經用過,我們就要跳過這次迴圈,獲得下一個物品的編號。 若這項檢查通過的話,我們要進行以下動作:將這樣物品標示為已用過,再進行下一層的 DFS,而如果下一層的 DFS 已經結束並回到這層,記得要先將這層所使用的物品重設為沒有使用過,再進行下一個 for 迴圈或回到上一層 DFS。
說了這麼多,好像還是沒有好好的解釋為什麼 DFS 能夠勝任這樣的工作,這裡我就用一張圖來代表剛剛所做的全部事情吧,相信會突然變得很好懂。
其實我們剛剛就是在對這個樹狀圖做 DFS,每次無路可走時,就代表找到了一種解法。應該滿好理解的吧?
那我們要進入下一個問題囉:要怎麼處理比較特別的排列?比如說物品不限使用次數的排列,或著有相同物的排列。
物品不限使用次數的排列,也就是上述的遊戲中,沒有限定三人一定要出不同的拳,想怎麼出就怎麼出。這種的比較簡單,也許你已經知道了,就是不要使用 visited[] 來記錄使用情況嘛!沒錯,所以我們只要把跟 visited[] 有關的部分全部砍掉就好了。
接下來的比較複雜(其實也還好):只能用特定數量來排列的有相同物之排列。
- 剪刀、石頭、布!-全部排列
現在有三個人:甲、乙、丙在猜拳,已知他們會出完全不同的拳,而你想知道依照甲、乙、丙的順序,他們出拳的排列有哪幾種的話,就可以用 DFS 來算喔!
...蛤?用 DFS?這不是樹的走訪嗎?管他的,先上程式碼!
- C++
string gesture[3] = {"剪刀", "石頭", "布"};
bool visited[3] = {false};
string arrangement[3];
void dfs(int layer){
if (layer == 3){
for (int i = 0; i < 3; i++){
cout << arrangement[i] << "\t";
}
cout << endl;
return;
}
for (int i = 0; i < 3; i++){
if (visited[i]){
continue;
}
visited[i] = true;
arrangement[layer] = gesture[i];
dfs(layer + 1);
visited[i] = false;
}
}
- Python
gesture = ['剪刀', '石頭', '布']
visited = [False] * 3
arrangement = [''] * 3
def dfs(layer):
if layer == 3:
print(*arrangement, sep='\t\t')
return
for i in range(3):
if visited[i]:
continue
visited[i] = True
arrangement[layer] = gesture[i]
dfs(layer + 1)
visited[i] = False
完整請見:C++ / Python輸出:
剪刀 石頭 布
剪刀 布 石頭
石頭 剪刀 布
石頭 布 剪刀
布 剪刀 石頭
布 石頭 剪刀
- 解密時間!
根據這樣的想法,我們可以發現 DFS 可以很容易的解決這個問題:將一位數字交由一層 DFS 來處理,而在每一層 DFS 中,我們要做以下這些事情:
1) 檢查是不是最後一層,如果是的話,代表已經得到一組排列,應將結果輸出,並回到上一層
2) 選定一個數字(代表欲排列物之編號),進行下一層的 DFS
讓我們看看 2) 這項敘述的詳細內容:首先是選一個數字,由於我們想讓每樣物品都輪過一次,所以我們使用 for 迴圈來依序取得編號。 接下來很重要的一點,因為有些東西我們可能已經排在前面了,所以我們要用一個陣列 visited[] 來儲存是否用過,如果已經用過,我們就要跳過這次迴圈,獲得下一個物品的編號。 若這項檢查通過的話,我們要進行以下動作:將這樣物品標示為已用過,再進行下一層的 DFS,而如果下一層的 DFS 已經結束並回到這層,記得要先將這層所使用的物品重設為沒有使用過,再進行下一個 for 迴圈或回到上一層 DFS。
說了這麼多,好像還是沒有好好的解釋為什麼 DFS 能夠勝任這樣的工作,這裡我就用一張圖來代表剛剛所做的全部事情吧,相信會突然變得很好懂。
那我們要進入下一個問題囉:要怎麼處理比較特別的排列?比如說物品不限使用次數的排列,或著有相同物的排列。
物品不限使用次數的排列,也就是上述的遊戲中,沒有限定三人一定要出不同的拳,想怎麼出就怎麼出。這種的比較簡單,也許你已經知道了,就是不要使用 visited[] 來記錄使用情況嘛!沒錯,所以我們只要把跟 visited[] 有關的部分全部砍掉就好了。
接下來的比較複雜(其實也還好):只能用特定數量來排列的有相同物之排列。
- 發放糖果-有相同物的排列
乍看之下好像有點複雜,因為要考慮那兩顆紅色的糖果,就算交換位置也還是一樣的分法,感覺好像要刪去一些分法,也很麻煩。這時可能會想說「畫個樹狀圖好了」,結果畫出來... 參差不齊的,根本很難對它做 DFS。
這時候,我們只要換一下想法就好,找出剛剛那個邏輯的不適合處加以替換。很顯然的,是我們用來記錄有沒有用過的 visited[] 陣列出了問題,因為對於紅色糖果來說,用掉一顆並不算真正的用完了,於是在這裡就會出現一些問題。那我們該怎麼處理呢?這裡就交給讀者們想囉。
*參考答案:C++ / Python
最後我們來思考一下,n 物取 k 物的排列方法,也就是高中所學的 P 幾取幾。
首先是 visited[] 陣列,既然剛剛的案例似乎都是它出問題,這次也是嗎?不,這次不是它的問題,因為甲~戊這五人都是唯一的,所以並沒有違反當初我們使用 visited[] 的用意。那麼應該也只剩下一個地方有疑點了:判斷是否為最後一層的地方。一直都沒思考這地方 if 判斷式裡的數字是怎麼來的,因為之前是有幾個要排就填多少,但這次我們只要取前三個,所以數字就填 3 吧。(沒錯,就是這樣)
先看程式碼應該比較好懂:
可以看到雖然有五個人在輪,但 if 判斷的地方只要跑 3 層 DFS 就會停止,這樣就可以達到我們只要取前三名的目的了。
1) 下圖是最後這個程式執行結果的一部分,試問若把 if 判斷中的數字改為 2 的話,會不會造成輸出中有三組「甲 乙」、「甲 丙」...等情況,也就是只有下圖中的第三直行消失而已?為什麼會 / 不會?
*參考答案:C++ / Python
最後我們來思考一下,n 物取 k 物的排列方法,也就是高中所學的 P 幾取幾。
- 誰是前三名?-n 物取 k 物的排列
首先是 visited[] 陣列,既然剛剛的案例似乎都是它出問題,這次也是嗎?不,這次不是它的問題,因為甲~戊這五人都是唯一的,所以並沒有違反當初我們使用 visited[] 的用意。那麼應該也只剩下一個地方有疑點了:判斷是否為最後一層的地方。一直都沒思考這地方 if 判斷式裡的數字是怎麼來的,因為之前是有幾個要排就填多少,但這次我們只要取前三個,所以數字就填 3 吧。(沒錯,就是這樣)
先看程式碼應該比較好懂:
- C++
bool visited[5] = {false};
int arrangement[3];
string people[5] = {"甲", "乙", "丙", "丁", "戊"};
void dfs(int layer){
if (layer == 3){
for (int j = 0; j < 3; j++){
cout << people[arrangement[j]] << " ";
} cout << endl;
return;
}
for (int i = 0; i < 5; i++){
if (visited[i]){
continue;
}
visited[i] = true;
arrangement[layer] = i;
dfs(layer + 1);
visited[i] = false;
}
}
- Python
visited = [False] * 5
arrangement = [''] * 3
people = ['甲', '乙', '丙', '丁', '戊']
def dfs(layer):
if layer == 3:
print(*arrangement, sep=' ')
return
for i in range(5):
if visited[i]:
continue
visited[i] = True
arrangement[layer] = people[i]
dfs(layer + 1)
visited[i] = False
完整請見:C++ / Python可以看到雖然有五個人在輪,但 if 判斷的地方只要跑 3 層 DFS 就會停止,這樣就可以達到我們只要取前三名的目的了。
- 動腦時間!
1) 下圖是最後這個程式執行結果的一部分,試問若把 if 判斷中的數字改為 2 的話,會不會造成輸出中有三組「甲 乙」、「甲 丙」...等情況,也就是只有下圖中的第三直行消失而已?為什麼會 / 不會?
2) 你能簡單的歸納出 DFS 的步驟嗎?
3) 能夠使用 DFS 列出 n 物取 k 物的組合種類嗎?如果可以,請實作看看;如果不行,試想想為什麼以及如何用別的方法解決。
後記:
終於打完這一篇了,看了看上次修改日期,嗯...原來是去年6月就應該要產出的啊,我又在耍廢了XD
至於後面的題目,算是一時興起的吧(欸),列出了我在寫這篇時突然想到的一些問題,可是我明明記得還有很多問題的啊,看來寫一寫就自動消失了呢,真不錯(是嗎
希望中間那個留給讀者思考不會被發現是我懶得打字。
留言
張貼留言