 870e3e5cb2
			
		
	
	870e3e5cb2
	
	
	
		
			
			* Sync zh and zh-hant versions * Update en/README.md * Add a Q&A for chapter of introduction * Update the callout headers * Sync zh ang zh-hant versions * Bug fixes
		
			
				
	
	
	
		
			5.6 KiB
		
	
	
	
	
	
	
	
			
		
		
	
	全排列問題
全排列問題是回溯演算法的一個典型應用。它的定義是在給定一個集合(如一個陣列或字串)的情況下,找出其中元素的所有可能的排列。
下表列舉了幾個示例資料,包括輸入陣列和對應的所有排列。
表 全排列示例
| 輸入陣列 | 所有排列 | 
|---|---|
| [1] | [1] | 
| [1, 2] | [1, 2], [2, 1] | 
| [1, 2, 3] | [1, 2, 3], [1, 3, 2], [2, 1, 3], [2, 3, 1], [3, 1, 2], [3, 2, 1] | 
無相等元素的情況
!!! question
輸入一個整數陣列,其中不包含重複元素,返回所有可能的排列。
從回溯演算法的角度看,我們可以把生成排列的過程想象成一系列選擇的結果。假設輸入陣列為 [1, 2, 3] ,如果我們先選擇 1 ,再選擇 3 ,最後選擇 2 ,則獲得排列 [1, 3, 2] 。回退表示撤銷一個選擇,之後繼續嘗試其他選擇。
從回溯程式碼的角度看,候選集合 choices 是輸入陣列中的所有元素,狀態 state 是直至目前已被選擇的元素。請注意,每個元素只允許被選擇一次,因此 state 中的所有元素都應該是唯一的。
如下圖所示,我們可以將搜尋過程展開成一棵遞迴樹,樹中的每個節點代表當前狀態 state 。從根節點開始,經過三輪選擇後到達葉節點,每個葉節點都對應一個排列。
重複選擇剪枝
為了實現每個元素只被選擇一次,我們考慮引入一個布林型陣列 selected ,其中 selected[i] 表示 choices[i] 是否已被選擇,並基於它實現以下剪枝操作。
- 在做出選擇 choice[i]後,我們就將selected[i]賦值為\text{True},代表它已被選擇。
- 走訪選擇串列 choices時,跳過所有已被選擇的節點,即剪枝。
如下圖所示,假設我們第一輪選擇 1 ,第二輪選擇 3 ,第三輪選擇 2 ,則需要在第二輪剪掉元素 1 的分支,在第三輪剪掉元素 1 和元素 3 的分支。
觀察上圖發現,該剪枝操作將搜尋空間大小從 O(n^n) 減小至 O(n!) 。
程式碼實現
想清楚以上資訊之後,我們就可以在框架程式碼中做“完形填空”了。為了縮短整體程式碼,我們不單獨實現框架程式碼中的各個函式,而是將它們展開在 backtrack() 函式中:
[file]{permutations_i}-[class]{}-[func]{permutations_i}
考慮相等元素的情況
!!! question
輸入一個整數陣列,**陣列中可能包含重複元素**,返回所有不重複的排列。
假設輸入陣列為 [1, 1, 2] 。為了方便區分兩個重複元素 1 ,我們將第二個 1 記為 \hat{1} 。
如下圖所示,上述方法生成的排列有一半是重複的。
那麼如何去除重複的排列呢?最直接地,考慮藉助一個雜湊集合,直接對排列結果進行去重。然而這樣做不夠優雅,因為生成重複排列的搜尋分支沒有必要,應當提前識別並剪枝,這樣可以進一步提升演算法效率。
相等元素剪枝
觀察下圖,在第一輪中,選擇 1 或選擇 \hat{1} 是等價的,在這兩個選擇之下生成的所有排列都是重複的。因此應該把 \hat{1} 剪枝。
同理,在第一輪選擇 2 之後,第二輪選擇中的 1 和 \hat{1} 也會產生重複分支,因此也應將第二輪的 \hat{1} 剪枝。
從本質上看,我們的目標是在某一輪選擇中,保證多個相等的元素僅被選擇一次。
程式碼實現
在上一題的程式碼的基礎上,我們考慮在每一輪選擇中開啟一個雜湊集合 duplicated ,用於記錄該輪中已經嘗試過的元素,並將重複元素剪枝:
[file]{permutations_ii}-[class]{}-[func]{permutations_ii}
假設元素兩兩之間互不相同,則 n 個元素共有 n!  種排列(階乘);在記錄結果時,需要複製長度為 n 的串列,使用 O(n) 時間。因此時間複雜度為 $O(n!n)$ 。
最大遞迴深度為 n ,使用 O(n) 堆疊幀空間。selected 使用 O(n) 空間。同一時刻最多共有 n 個 duplicated ,使用 O(n^2) 空間。因此空間複雜度為 $O(n^2)$ 。
兩種剪枝對比
請注意,雖然 selected 和 duplicated 都用於剪枝,但兩者的目標不同。
- 重複選擇剪枝:整個搜尋過程中只有一個 selected。它記錄的是當前狀態中包含哪些元素,其作用是避免某個元素在state中重複出現。
- 相等元素剪枝:每輪選擇(每個呼叫的 backtrack函式)都包含一個duplicated。它記錄的是在本輪走訪(for迴圈)中哪些元素已被選擇過,其作用是保證相等元素只被選擇一次。
下圖展示了兩個剪枝條件的生效範圍。注意,樹中的每個節點代表一個選擇,從根節點到葉節點的路徑上的各個節點構成一個排列。




