输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)

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;

    }

}
*/
import java.util.*;
public class Solution {
    
    
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root, int target){
        ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();
        //Stack<TreeNode> stack = new Stack<TreeNode>();
        if(root==null){
            return list;
        }
        ArrayList<Integer> temp = new ArrayList<>();
        
        findResultPath(root, list, temp, target);
        return list;
    }

    private void findResultPath(TreeNode node, ArrayList<ArrayList<Integer>> list, ArrayList<Integer> temp, int target) {
        temp.add(node.val);
        target -= node.val;
        if(node.left == null && node.right == null){
            if(target == 0){
                list.add(new ArrayList<Integer>(temp));
            }
                
            temp.remove(temp.size()-1);
            return;
        }
        if(node.left != null){
             findResultPath(node.left, list, temp, target);
        }
           
        if(node.right != null){
            findResultPath(node.right, list, temp, target);
        }
            
        temp.remove(temp.size()-1);
    }
}

猜你喜欢

转载自www.cnblogs.com/q-1993/p/10726412.html