【Leetcode】113. 路徑總和II

題目

給定一個二叉樹和一個目標和，找到全部從根節點到葉子節點路徑總和等於給定目標和的路徑。

說明: 葉子節點是指沒有子節點的節點。

示例:
給定以下二叉樹，以及目標和 sum = 22，

5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1

返回:

[
   [5,4,11,2],
   [5,8,4,5]
]

題解

這道題目是上一道的延伸，可是須要記錄下路徑，返回回去。這就是一個典型的backtrack的題目了。咱們用迭代的方式須要記錄中間的路徑狀態，稍顯複雜，因此咱們想用遞歸的方式來解，先探索左子樹，而後探索右子樹。若是都探索完以後，右知足的就加入到最終結果中。

public class Solution {
    public List<List<Integer>> pathSum(TreeNode root, int sum) {
        List<List<Integer>> res = new LinkedList<>();
        helper(root, sum, res, new LinkedList<>());
        return res;
    }

    public void helper(TreeNode root, int sum, List<List<Integer>> res, List<Integer> current) {
        if (root == null) {
            return;
        }
        current.add(root.val);
        if (root.left == null && root.right == null && sum == root.val) {
            // leaf node.
            res.add(new LinkedList<>(current));
            // back track.
            current.remove(current.size() - 1);
            return;
        }

        helper(root.left, sum - root.val, res, current);
        helper(root.right, sum - root.val, res, current);
        // back track.
        current.remove(current.size() - 1);
    }
}

熱門閱讀

• 技術文章彙總
• 【Leetcode】103. 二叉樹的鋸齒形層次遍歷
• 【Leetcode】102. 二叉樹的層次遍歷
• 【Leetcode】101. 對稱二叉樹
• 【Leetcode】100. 相同的樹
• 【Leetcode】98. 驗證二叉搜索樹

手撕代碼QQ羣：805423079， 羣密碼：1024