40:二叉树中和为某一值的路径

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

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

public class Offer40 {
   public static void main(String[] args) {
       TreeNode node = Main.createNode();
       new Offer40().FindPath(node, 7);
   }

   public ArrayList<ArrayList<Integer>> FindPath(TreeNode root, int target) {
       ArrayList<ArrayList<Integer>> listResult = new ArrayList<ArrayList<Integer>>();

       if (root == null || root.value > target)
           return listResult;

       ArrayList<Integer> list = new ArrayList<Integer>();

       findPath(root, target, listResult, list);

       System.out.println(listResult);

       return listResult;

   }

   public void findPath(TreeNode root, int target, ArrayList<ArrayList<Integer>> listResult, ArrayList<Integer> list) {
       //如果当前结点为空 或者 当前结点值大于target 清空路径
       if (root == null || root.value > target) {
           //清空该路径
           //Removes all of the elements from this list.
           list.clear();
       }

       //如果当前结点值等于target && 是叶子节点  将该结点添加到list中 否则清除
       else if (root.value == target) {
           if (root.left == null && root.right == null) {
               list.add(root.value);
               listResult.add(list);
           } else {
               list.clear();
           }
       }

       //如果当前结点值小于target,递归找他的左右子树
       else {
           list.add(root.value);
           ArrayList<Integer> list2 = new ArrayList<Integer>();

           list2.addAll(list);
           target -= root.value;

           findPath(root.left, target, listResult, list);
           findPath(root.right, target, listResult, list2);

       }
   }

}

已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 书香水墨 设计师:CSDN官方博客 返回首页