二叉树中和为某一值的路径

题目

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

思路

深度优先搜索。使用前序遍历，使用两个全局变量result和tmp，result来存放最终结果，tmp用来存放临时结果。

每次遍历，我们先把root的值压入tmp，然后判断当前root是否同时满足：

1 与给定数值相减为0；

2 左子树为空；

3 右子树为空。

如果满足条件，就将tmp压入result中，否则，依次遍历左右子树。需要注意的是，遍历左右子树的时候，全局变量tmp是不清空的，直到到了根结点才请空tmp。

代码一

public static void main(String[] args) {
        TreeNode root1 = new TreeNode(10);
        TreeNode node1 = new TreeNode(5);
        TreeNode node2 = new TreeNode(12);
        TreeNode node3 = new TreeNode(4);
        TreeNode node4 = new TreeNode(7);
//        TreeNode node5 = new TreeNode(6);
//        TreeNode node6 = new TreeNode(7);
        root1.left=node1;
        root1.right=node2;
        node1.left =node3;
        node1.right = node4;
//        node2.left = node5;
//        node2.right = node6;
        System.out.println(pathSum(root1,22));
    }
    public static List<List<Integer>> paths = new ArrayList<>();
    public static List<List<Integer>> pathSum(TreeNode root, int sum) {
        go(root,0,sum,new ArrayList<>());
        return paths;
    }

    public static void go(TreeNode node,int sum,int target,ArrayList<Integer> path){
        if(node==null){
            return;
        }
        sum+=node.val;
        path.add(node.val);
        //判断该路径和是否符合目标值
        if(sum==target&&node.left==null&&node.right==null){
            //因为要子节点，所以它的左子树和右子树必须为null
            paths.add(new ArrayList<>(path));
        }else{
            go(node.left,sum,target,path);
            go(node.right,sum,target,path);
        }
        //回溯
        path.remove(path.size()-1);
    }