回溯法-獲取path set,一般采用樹結構解題

????回溯實際上就是遍歷的變種,不符合條件時,本次遍歷向上回退。一般來說,回溯算法都可以將決策路徑畫成樹的形狀,成為一棵搜索樹?;厮莘▓绦械倪^程實際上就是在這棵樹上做遍歷。使用回溯法的題目,為什么不能用遞歸法,因為回溯法中記錄路徑的棧只有一個。

1、回溯算法的基本思想

????回溯算法的定義:回溯法采用試錯的思想,當它通過嘗試發現現有的分步答案不能得到有效的正確的解答的時候,它將取消上一步甚至是上幾步的計算,再通過其它的可能的分步解答再次嘗試尋找問題的答案?!?回溯法 - 維基百科[3]

????從字面意思上來看,回溯(backtracking) 實際上就是“撤回一步”的意思。而在二叉樹的 DFS 遍歷中,從一個結點退出就是一種回溯?;厮莘ê?DFS 是息息相關的。

????根據回溯操作的特性,我們使用棧記錄遍歷時的當前路徑。當進入一個結點時,做 push 操作;當退出一個結點時,做 pop 操作,進行回溯。

2、案例1

????給定一個二叉樹和一個目標和,找到所有從根結點到葉結點的路徑,使得路徑上所有結點值相加等于目標和。

public List<List<Integer>> pathSum(TreeNode root, int sum) {

? ? List<List<Integer>> res = new ArrayList<>();

? ? Deque<Integer> path = new ArrayDeque<>();

? ? traverse(root, sum, path, res);

? ? return res;

}

void traverse(TreeNode root, int sum, Deque<Integer> path, List<List<Integer>> res) {

? ? if (root == null) {

? ? ? ? return;

? ? }

? ? path.addLast(root.val);

? ? if (root.left == null && root.right == null) {

? ? ? ? if (root.val == sum) {

? ? ? ? ? ? res.add(new ArrayList<>(path));

? ? ? ? }

? ? }

? ? int target = sum - root.val;

? ? traverse(root.left, target, path, res);

? ? traverse(root.right, target, path, res);

? ? path.removeLast();

}

????代碼的整體結構和上期例題題解類似,只是加上了棧 path 記錄當前路徑。關于棧的 push 和 pop 操作,有兩個需要注意的地方:

????????????* 保證剛進入結點就 push,最后退出結點之前才 pop,這樣才能使當前路徑和遍歷的進度對應;

????????????* 在葉結點判斷后,不能進行 return,否則會跳過后面的 pop 操作而出錯。

????這兩點都需要做題來體驗,建議親自做一遍例題來體會。

3、案例2

? ? 題目:給定一組不含重復元素的整數數組 nums,返回該數組所有可能的子集(冪集)。

????Subsets 問題就是要枚舉出集合的所有子集。生成子集有一個很簡單的策略,一個子集可以選擇使用或不使用第一個元素,選好之后,再對第二個元素進行選擇,以此類推。這就是一種回溯的思想。這又是一個樹的結構。一般來說,回溯算法都可以將決策路徑畫成樹的形狀,成為一棵搜索樹?;厮莘▓绦械倪^程實際上就是在這棵樹上做遍歷。剛好這還是一棵二叉樹,這又聯系上了二叉樹的遍歷。

????那么,我們可以嘗試用遍歷樹的思路寫出回溯法的代碼。這里的棧是當前子集里的元素,push 操作是往子集里加元素,pop 操作是從子集中刪除元素(撤銷選擇)。

????最終我們得到完整的代碼:

public List<List<Integer>> subsets(int[] nums) {

? ? Deque<Integer> current = new ArrayDeque<>(nums.length);

? ? List<List<Integer>> res = new ArrayList<>();

? ? backtrack(nums, 0, current, res);

? ? return res;

}

void backtrack(int[] nums, int k, Deque<Integer> current, List<List<Integer>> res) {

? ? if (k == nums.length) {

? ? ? ? res.add(new ArrayList<>(current));

? ? ? ? return;

? ? }

? ? // 不選擇第 k 個元素

? ? backtrack(nums, k+1, current, res);

? ? // 選擇第 k 個元素

? ? current.addLast(nums[k]);

? ? backtrack(nums, k+1, current, res);

? ? current.removeLast();

}

????這份代碼看起來和 Path Sum II 的代碼非常類似,例如都使用了一個棧,遞歸的參數也很像。但是遞歸調用和 push/pop 的操作方式有一些微妙的地方。

????現在,我們是在調用遞歸函數之前和之后進行 push/pop,這是因為數組本身并沒有遞歸結構,我們需要用 push/pop 操作來營造出不同的選擇。兩個遞歸函數的調用其實都是一樣的,但因為 current 中的內容不一樣,所以其實是兩個決策路徑。

4、時間復雜度

????回溯算法的復雜度一般都會很高。以 Subsets 問題為例,從搜索樹的規模可以看出算法的時間復雜度是非常高的 。不過,回溯法寫成這樣的復雜度是可接受的,一般的回溯法題目也沒有更高效的解法。

5、總結

????通過這兩個例題我們看到了回溯算法和二叉樹遍歷的相似關系。在求解回溯算法的時候,我們可以先構造一個搜索樹,在這個樹上遍歷進行遞歸求解。

????需要注意的是,例題 Subsets 中的搜索樹是二叉樹,這只是個巧合。實際上搜索樹完全可以是多叉樹,而且多叉樹才更常見。

????本篇講解的是比較基礎的回溯法思想?;厮莘ㄟ€有很多技巧,例如 Permutation 和 Combination 系列題目,后續還會有文章進行講解。

6、相關題目

????二叉樹遍歷的題目(理解遍歷思想):

????????????* 129 - Sum Root to Leaf Numbers[4]

????????????* 257 - Binary Tree Paths[5]

????回溯法題目(這里只列出比較簡單的兩道,更多的題目可以在 LeetCode 上尋找 backtracking 標簽):

????????????* 22 - Generate Parentheses[6]

????????????* 39 - Combination Sum[7]

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

推薦閱讀更多精彩內容

  • 單例定義:保證一個類僅有一個實例,并提供一個訪問它的全局訪問點。餓漢模式public class Singleto...
    小楊不想努力了閱讀 397評論 0 4
  • 回溯backtracking 回溯法思路的簡單描述是:把問題的解空間轉化成了圖或者樹的結構表示,然后使用深度優先搜...
    sylvainwang閱讀 2,274評論 0 51
  • Remove time complexity: remove from a set is O(1), remove...
    云端漫步_b5aa閱讀 642評論 0 0
  • 似乎,,,,返回結果的空間不會算在空間復雜度上 思維的全面性(各種邊界條件) 索引與中點的關系:設最后元素的索引為...
    大海一滴寫字的地方閱讀 526評論 1 0
  • day4 33.二叉搜索樹的后序遍歷序列 思路:運用遞歸,不斷判斷左右子樹的后序遍歷序列(最后一個數字是根節點,前...
    IAmKepler閱讀 244評論 0 0