剑指offer题解 二叉树中和为某一值的路径

题目描述

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

用两个ArrayList来实现，代码比较复杂。递归结束前，移除当前层级遍历时添加的元素

import java.util.ArrayList;
/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
        ArrayList<ArrayList<Integer>> mStack=new ArrayList<ArrayList<Integer>>();
        if(root==null) return mStack;
        ArrayList<Integer> temp=new ArrayList<Integer>();
        find(root,target,0,mStack,temp);
        return mStack;
    }
    void find(TreeNode root,int target,int sum , ArrayList<ArrayList<Integer>> mStack,ArrayList<Integer> temp ){
        temp.add(root.val);
        if(root.left==null&&root.right==null&&(sum+root.val==target)){
            ArrayList<Integer> res=new ArrayList<Integer>();
            for(int i = 0;i < temp.size(); i ++){
               res.add(temp.get(i));
            }
            mStack.add(res);
        }
        if(root.left!=null){
            find(root.left,target,sum+root.val,mStack,temp);
        }
        if(root.right!=null){
             find(root.right,target,sum+root.val,mStack,temp);
        }
        temp.remove(temp.size()-1);
    }
}