题目:输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
思想:由于本题中的根要在前面,因而我们首先想到的是前序遍历。当用前序遍历的方式访问某一节点时,我们把该节点添加到路径上,并累加该节点的值。如果该节点是叶节点并且路径中的节点值的和刚好为输入的整数,则就把此路径打印出来。否则继续遍历叶节点。当前节点的访问结束后,要回到他的父节点,因此在函数退出之前要在路径上删除当前节点并且减去当前节点的值,以确保回退时刚好是根节点到父节点的路径。
java代码:
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>> listAll= new ArrayList<ArrayList<Integer>>(); public ArrayList<Integer> list=new ArrayList<Integer>(); public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) { if(root==null){ return listAll; } list.add(root.val); target-=root.val; if(target==0 && root.left==null && root.right==null){ listAll.add(new ArrayList<Integer>(list)); } FindPath(root.left,target); FindPath(root.right,target); list.remove(list.size()-1); return listAll; } }