剑指Offer-二叉树中和为某一值的路径

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

输入一棵二叉树和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。

从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。

样例

给出二叉树如下所示，并给出num=22。
      5
     / \
    4   6
   /   / \
  12  13  6
 /  \    / \
9    1  5   1

输出：[[5,4,12,1],[5,6,6,5]]

思路

首先得理解题目的意思，要不这道题会很难做。

• 先来看“路径”，指的是从根节点到叶子节点，不是随便几个节点就能形成的路径。一定要以根节点为首，叶子节点为尾所形成的路径。
• 既然是找符合条件的路径，那么可能会“一不小心”多走了，或者走错了。这时候需要回溯到上一个状态，也就是DFS里的backTracking思想。
• 我们需要一个容器来保存中间结果，一旦遇到符合条件的路径终点，就要把容器里的元素全部加入到最终结果容器中。如果不符合条件，当回溯到上一个状态的时候，需要把当前状态已经存储的元素删除掉，一面影响后续的其他状态。

class Solution {
    //结果集
    private List<List<Integer>> res=new ArrayList<>();
    //中间结果集
    private ArrayList<Integer> mres=new ArrayList<>();
    public List<List<Integer>> findPath(TreeNode root, int sum) {      
        backTracking(root,sum,mres);
        return res;
    }
    public void backTracking(TreeNode root,int sum,ArrayList<Integer> mres)
    {
        if(root==null) return;
        //先加入到中间结果集中,不符合再拿出来,反正先加进去,符合就找到一个路径
        mres.add(root.val);
        sum-=root.val;
        //注意此处要符合题意,即必须是根节点,所以它的左右孩子为空才行
        if(sum==0 && root.left==null&&root.right==null)
        {
            //符合条件,填充到最终结果集中
            res.add(new ArrayList(mres));
        }else
        {
            //只要不符合条件,递归去左右子树中找
            backTracking(root.left,sum,mres);
            backTracking(root.right,sum,mres);
        }
        //如果不符合条件,一定要把当前状态消除
        mres.remove(mres.size()-1);
    }
}