LeetCode 113 路径总和II 题解

LeetCode链接

给定一个二叉树和一个目标和，找到所有从根节点到叶子节点路径总和等于给定目标和的路径。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树，以及目标和 sum = 22，

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1
返回:
[
   [5,4,11,2],
   [5,8,4,5]
]

方法：DFS

时间复杂度：O(N^2)，其中 N 是树的节点数。在最坏情况下，树的上半部分为链状，下半部分为完全二叉树，并且从根节点到每一个叶子节点的路径都符合题目要求。此时，路径的数目为 O(N)，并且每一条路径的节点个数也为 O(N)，因此要将这些路径全部添加进答案中，时间复杂度为 O(N^2)
空间复杂度：O(N)，其中 N 是树的节点数。空间复杂度主要取决于栈空间的开销，栈中的元素个数不会超过树的节点数。

注意，这道题没有说结点的值全为正数，也就是可能会出现负数，也就无法根据节点的值和sum的大小关系进行剪枝

class Solution {
    //频繁的尾部操作，用LinkedList更省时间
    List<List<Integer>> result = new LinkedList<>();
    LinkedList<Integer> tmp = new LinkedList<>();
    public List<List<Integer>> pathSum(TreeNode root, int sum) {
        if(root == null){
            return result;
        }
        dfs(root, sum);
        return result;
    }
    public void dfs(TreeNode root, int target){
        //剪枝...不大行，因为没说二叉树结点的值全是正数emm
        // if(root.val>target){
        //     return;
        // }
        tmp.add(root.val);
        target -= root.val;
        if(root.left == null && root.right == null && target == 0){
            //这里不能直接返回，因为tmp还需要回溯
            result.add(new ArrayList<Integer>(tmp));
        }
        if(root.left != null){
            dfs(root.left, target);
        }
        if(root.right != null){
            dfs(root.right, target);
        }
        //回溯
        tmp.removeLast();
    }
}