【追求进步】二叉树中和为某一值的路径

题目描述

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

在线代码:

import java.util.ArrayList;
import java.util.Stack;
/**
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>> pathlist=new ArrayList<ArrayList<Integer>>();
        if(root==null){
            return pathlist;
        }
        Stack<Integer> stack=new Stack<Integer>();//借助辅助栈
        FindPath(root,target,stack,pathlist);
        return pathlist;
    }
    private void FindPath(TreeNode root,int target,Stack<Integer> path,
                          ArrayList<ArrayList<Integer>> pathlist){
        
        if(root==null){
            return;
        }
        //遍历最终的结果处理
        if(root.left==null&&root.right==null){//走到叶子结点或者只有一个根结点的
            if(root.val==target){
                ArrayList<Integer> list= new ArrayList<Integer>();
                for(int i:path){
                    list.add(new Integer(i));
                }
                list.add(new Integer(root.val));//加入list
                pathlist.add(list);
            }
        }else{
            path.push(new Integer(root.val));//将元素入栈
            FindPath(root.left, target-root.val, path, pathlist);//左边递归
            FindPath(root.right, target-root.val, path,  pathlist);//右边递归
            path.pop();
        }
    }
    
    
}