[LeetCode]（面试题34）二叉树中和为某一值的路径

题目

输入一棵二叉树和一个整数，打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。

示例:
给定如下二叉树，以及目标和 sum = 22，

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

返回:

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

提示：

1. 节点总数 <= 10000

解题思路

主要思路：通过先序遍历，记录从根节点到当前节点的路径。当遍历到叶节点且路径上各节点值的和等于目标值 sum 时，将此路径加入结果列表。

递归步骤：

• 1、路径更新： 更新当前路径和，并将当前节点值 root.val 加入路径 path ；；
• 2、添加路径至结果列表：若 1）root 为叶节点；2）当前路径和等于目标值。两个条件同时满足，则将此路径 path 加入 res。
• 3、递归遍历子节点： 如果左、右节点非空，则递归遍历左、右子节点。
• 4、回溯：返回父节点时，需要将当前节点从路径 path 中删除。

复杂度分析：
时间复杂度：O(N)。N 为二叉树的节点数，先序遍历需要遍历所有节点。
空间复杂度：O(N)。最差情况下，即树退化为链表时，path 存储所有树节点，使用 O(N) 额外空间。

注：
1）记录路径时若直接执行 res.add(path) ，则是将 path 对象加入了 res。后续 path 改变时， res 中的 path 对象也会随之改变。
2）正确做法：res.add(new LinkedList(path)) ，相当于复制了一个 path 并加入到 res 。

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    private List<List<Integer>> res = new ArrayList<>();

    public List<List<Integer>> pathSum(TreeNode root, int sum) {
        if(root == null){
            return res;
        }
        LinkedList<Integer> path = new LinkedList<>();
        dfs(root, path, 0, sum);
        return res;
    }

    private void dfs(TreeNode root, LinkedList<Integer> path, int curSum, int sum){
        curSum += root.val;
        path.offerLast(root.val);
        // 如果是叶节点，并且路径和等于期望值，就加入结果列表
        if(curSum == sum && root.left == null && root.right == null){
            res.add(new LinkedList(path));
            // res.add(path);
        }
        if(root.left != null){
            dfs(root.left, path, curSum, sum);
        }
        if(root.right != null){
            dfs(root.right, path, curSum, sum);
        }
        // 回溯：返回父节点时，在路径上删除当前节点
        path.removeLast();
    }
}