回溯法-獲取path set,一般采用樹(shù)結(jié)構(gòu)解題

????回溯實(shí)際上就是遍歷的變種,不符合條件時(shí),本次遍歷向上回退。一般來(lái)說(shuō),回溯算法都可以將決策路徑畫(huà)成樹(shù)的形狀,成為一棵搜索樹(shù)?;厮莘▓?zhí)行的過(guò)程實(shí)際上就是在這棵樹(shù)上做遍歷。使用回溯法的題目,為什么不能用遞歸法,因?yàn)榛厮莘ㄖ杏涗浡窂降臈V挥幸粋€(gè)。

1、回溯算法的基本思想

????回溯算法的定義:回溯法采用試錯(cuò)的思想,當(dāng)它通過(guò)嘗試發(fā)現(xiàn)現(xiàn)有的分步答案不能得到有效的正確的解答的時(shí)候,它將取消上一步甚至是上幾步的計(jì)算,再通過(guò)其它的可能的分步解答再次嘗試尋找問(wèn)題的答案。—— 回溯法 - 維基百科[3]

????從字面意思上來(lái)看,回溯(backtracking) 實(shí)際上就是“撤回一步”的意思。而在二叉樹(shù)的 DFS 遍歷中,從一個(gè)結(jié)點(diǎn)退出就是一種回溯?;厮莘ê?DFS 是息息相關(guān)的。

????根據(jù)回溯操作的特性,我們使用棧記錄遍歷時(shí)的當(dāng)前路徑。當(dāng)進(jìn)入一個(gè)結(jié)點(diǎn)時(shí),做 push 操作;當(dāng)退出一個(gè)結(jié)點(diǎn)時(shí),做 pop 操作,進(jìn)行回溯。

2、案例1

????給定一個(gè)二叉樹(shù)和一個(gè)目標(biāo)和,找到所有從根結(jié)點(diǎn)到葉結(jié)點(diǎn)的路徑,使得路徑上所有結(jié)點(diǎn)值相加等于目標(biāo)和。

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();

}

????代碼的整體結(jié)構(gòu)和上期例題題解類(lèi)似,只是加上了棧 path 記錄當(dāng)前路徑。關(guān)于棧的 push 和 pop 操作,有兩個(gè)需要注意的地方:

????????????* 保證剛進(jìn)入結(jié)點(diǎn)就 push,最后退出結(jié)點(diǎn)之前才 pop,這樣才能使當(dāng)前路徑和遍歷的進(jìn)度對(duì)應(yīng);

????????????* 在葉結(jié)點(diǎn)判斷后,不能進(jìn)行 return,否則會(huì)跳過(guò)后面的 pop 操作而出錯(cuò)。

????這兩點(diǎn)都需要做題來(lái)體驗(yàn),建議親自做一遍例題來(lái)體會(huì)。

3、案例2

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

????Subsets 問(wèn)題就是要枚舉出集合的所有子集。生成子集有一個(gè)很簡(jiǎn)單的策略,一個(gè)子集可以選擇使用或不使用第一個(gè)元素,選好之后,再對(duì)第二個(gè)元素進(jìn)行選擇,以此類(lèi)推。這就是一種回溯的思想。這又是一個(gè)樹(shù)的結(jié)構(gòu)。一般來(lái)說(shuō),回溯算法都可以將決策路徑畫(huà)成樹(shù)的形狀,成為一棵搜索樹(shù)。回溯法執(zhí)行的過(guò)程實(shí)際上就是在這棵樹(shù)上做遍歷。剛好這還是一棵二叉樹(shù),這又聯(lián)系上了二叉樹(shù)的遍歷。

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

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

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 個(gè)元素

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

? ? // 選擇第 k 個(gè)元素

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

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

? ? current.removeLast();

}

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

????現(xiàn)在,我們是在調(diào)用遞歸函數(shù)之前和之后進(jìn)行 push/pop,這是因?yàn)閿?shù)組本身并沒(méi)有遞歸結(jié)構(gòu),我們需要用 push/pop 操作來(lái)營(yíng)造出不同的選擇。兩個(gè)遞歸函數(shù)的調(diào)用其實(shí)都是一樣的,但因?yàn)?current 中的內(nèi)容不一樣,所以其實(shí)是兩個(gè)決策路徑。

4、時(shí)間復(fù)雜度

????回溯算法的復(fù)雜度一般都會(huì)很高。以 Subsets 問(wèn)題為例,從搜索樹(shù)的規(guī)模可以看出算法的時(shí)間復(fù)雜度是非常高的 。不過(guò),回溯法寫(xiě)成這樣的復(fù)雜度是可接受的,一般的回溯法題目也沒(méi)有更高效的解法。

5、總結(jié)

????通過(guò)這兩個(gè)例題我們看到了回溯算法和二叉樹(shù)遍歷的相似關(guān)系。在求解回溯算法的時(shí)候,我們可以先構(gòu)造一個(gè)搜索樹(shù),在這個(gè)樹(shù)上遍歷進(jìn)行遞歸求解。

????需要注意的是,例題 Subsets 中的搜索樹(shù)是二叉樹(shù),這只是個(gè)巧合。實(shí)際上搜索樹(shù)完全可以是多叉樹(shù),而且多叉樹(shù)才更常見(jiàn)。

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

6、相關(guān)題目

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

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

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

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

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

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

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書(shū)系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

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

友情鏈接更多精彩內(nèi)容