面试题34. 二叉树中和为某一值的路径 dfs遍历

题目描述

输入一棵二叉树和一个整数，打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
示例:
给定如下二叉树，以及目标和 sum = 22，

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

返回:
[
[5,4,11,2],
[5,8,4,5]
]
提示：
节点总数 <= 10000
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/er-cha-shu-zhong-he-wei-mou-yi-zhi-de-lu-jing-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

dfs解法

思路

采用dfs方法，进行先序遍历。遍历树的叶子节点，找出从根节点到叶子节点的路径和，如果等于给定sum值，则加入path集合。

这里定义两个字段LinkedList<List<Integer>> paths和LinkedList<Integer> path，前者用来记录已搜索到的满足条件的路径，后者用来维护当前路径。

原始的dps遍历，搜索顺序不是按照二叉树相邻节点移动的。故搜素顺序不一定是一条连续路径。这里采用一个处理，实现了维护一条连续路径。在递推过程中，先将当前节点加入维护的路径链表中，最后在分别访问左、右子树后，将当前节点从维护的路径列表中弹出。

java代码

    private LinkedList<List<Integer>> paths = new LinkedList<>();
    private LinkedList<Integer> path = new LinkedList<>();

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

        dfs(root, sum);
        return paths;
    }

    public void dfs(TreeNode aRoot, int aSum){

        if(aRoot==null)
            return;
        
        int val = aRoot.val;
        path.add(val);

        if(aSum-val == 0 && aRoot.left==null && aRoot.right==null){
            paths.add(new LinkedList<Integer>(path));
        }

        dfs(aRoot.left, aSum-val);
        dfs(aRoot.right, aSum-val);

        path.removeLast();
    }