剑指Offer对答如流系列 - 二叉树中和为某一值的路径

面试题34：二叉树中和为某一值的路径

一、题目描述

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

比如说下面这棵树，输入二叉树和整数22，则打印出两条路径，一条是10 12 另一条是10 5 7

树的结构定义如下：

public class Node{
        int e;
        Node left;
        Node right;
        Node(int x) { e = x; }
    }

二、问题分析

从上往下，找一条路径，由于是从根节点出发，我们选用前序遍历。

如果我们找到了其中一条路径，达到叶节点后，由于没有指向父节点的引用，所以必须提前创建一个链表存储前面经过的节点。

基于上面可以说，通过前序遍历，从根节点出发，每次在链表中存储遍历到的节点，若到达叶子节点，则根据所有节点的和是否等于输入的整数，判断是否打印输出。

强调的这部分 可以变成 在每次遍历到一个节点时，令目标整数等于自己减去当前节点的值，若到达根节点时，最终的目标整数等于0就可以打印输出， 上面的是书上的，这里的是个人实现的。

在当前节点访问结束后，我们需要返回到它的父节点，这个可以通过递归来实现，不过在函数一层一层返回之前要删除链表中的当前节点，以确保最终正确返回父节点。

三、问题解答

    ArrayList<ArrayList<Integer>> res = new ArrayList<>();
    ArrayList<Integer> path = new ArrayList<>();
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
        if (root == null) {
            return res;
        }
        findPath(root, target, 0);
        return res;
    }

    public void findPath(TreeNode root, int target, int sum) {
        //因为FindPath中和 下面程序中都进行了判null操作，root绝对不可能为 null
        path.add(root.val);
        sum += root.val;
        //已经到达叶子节点，并且正好加出了target
        if (sum == target && root.left == null && root.right == null) {
            //将该路径加入res结果集中
            res.add(new ArrayList(path));
        }
        //如果左子树非空，递归左子树
        if (root.left != null) {
            findPath(root.left, target , sum);
        }
        //如果右子树非空，递归右子树
        if (root.right != null) {
            findPath(root.right, target ,sum);
        }
        //无论当前路径是否加出了target，必须去掉最后一个，然后返回父节点，去查找另一条路径，最终的path肯定为null
        path.remove(path.size() - 1);
        return;

    }