题目描述:
输入一棵二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根节点开始往下一直到叶节点所经过的结点形成一条路径。
思路:
1、由于路径是从根节点出发到叶节点,也就是说路径总是以根节点为起点,因此我们首先需要遍历根节点,在树的前序、中序和后序三种遍历方式中,只有前序遍历是首先访问根节点的。
2、当用前序遍历的方式访问某一结点时,我们把该节点添加到路径上,并累加该节点的值。如果该结点为叶节点并且路径上结点的值的和正好等于输入的整数,则当前的路径符合要求,我们把它打印出来。如果当前节点不是叶节点,则继续访问它的子节点。当前访问结束后,递归函数将自动回到它的父节点。因此我们在函数退出之前要在路径上删除当前节点并减去当前节点的值,以确保返回父节点时路径刚好是从根节点到父节点的路径。我们不难看出保存路径的数据结构实际上是一个栈,因为路径要与递归调用状态一致,而递归调用的本质就是一个压栈和出栈的过程。
代码:
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>> listPath=new ArrayList<ArrayList<Integer>>();
Stack<Integer> stack=new Stack<Integer>();
if(root==null){//树为空树
return listPath;
}
FindPath(root,target,stack,listPath);
return listPath;
}
public void FindPath(TreeNode root,int target,Stack<Integer> stack,ArrayList<ArrayList<Integer>> listPath){
if(root==null){
return;
}
if(root.left==null&&root.right==null){//是叶子节点
if(root.val==target){
ArrayList<Integer> list=new ArrayList<Integer>();
for(int i:stack){
list.add(new Integer(i));
}
list.add(new Integer(root.val));
listPath.add(list);
}
}
else{//非叶子节点
stack.push(new Integer(root.val));
FindPath(root.left,target-root.val,stack,listPath);
FindPath(root.right,target-root.val,stack,listPath);
stack.pop();
}
}
}
不使用栈,代码:
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 {
ArrayList<Integer> listNode=new ArrayList<Integer>();//存放节点
ArrayList<ArrayList<Integer>> listPath=new ArrayList<ArrayList<Integer>>();//存放路径
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
if(root==null){//树为空树
return listPath;
}
listNode.add(root.val);
target-=root.val;
if(target==0&&root.left==null&&root.right==null){
listPath.add(new ArrayList<Integer>(listNode));
}
FindPath(root.left,target);
FindPath(root.right,target);
listNode.remove(listNode.size()-1);//把链表中最后添加进去的节点删除掉
return listPath;
}
}
代码的主要思想是树的先序遍历
listNode.remove(listNode.size()-1):这是模拟了栈退回,当前节点为叶子节点或者已经访问过的情况下,回溯到父节点。
listPath.add(new ArrayList<Integer>(listNode)):必须要重新生成一个对象实例,并使用listNode对其进行初始化赋值,不能直接listPath.add(listNode),因为listNode本质上是引用,在各次遍历中会直接改变它的值,最后的路径集合中前面的路径也会被后面的覆盖。