题目:输入一个二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
/**
* 二叉树
* @author Administrator
*
*/
public class Node{
public Node left;
public Node right;
public int value;
public Node(){}
public Node(int value){this.value = value;}
}
/**
* 预处理边界条件
* @param phead
* @param expectedSum
*/
public void prepareFindPath(Node phead, int expectedSum){
if(phead == null)return;
int currentSum = 0;
Stack<Integer> stack = new Stack<Integer>();
findPath(phead, expectedSum, stack, currentSum);
}
/**
* 打印二叉树中和为某一值的路径
* @param node
* @param expectedSum
* @param list
* @param currentSum
*/
public void findPath(Node node, int expectedSum, Stack<Integer> stack, int currentSum){
currentSum += node.value;
stack.add(node.value);
if((currentSum == expectedSum) && node.left == null && node.right ==null){
System.out.println(Arrays.toString(stack.toArray()));
}
if(node.left != null)findPath(node.left, expectedSum, stack, currentSum);
if(node.right != null)findPath(node.right, expectedSum, stack, currentSum);
currentSum -= node.value;
stack.pop();
}
总结:
当用前序遍历的方式访问到某一结点时,我们把该结点添加到路径上,并累加该结点的值。如果该结点为叶结点并且路径中结点值的和刚好等于输入的整数,则当前的路径符合要求,我们把它打印出来。如果当前结点不是叶节点,则继续访问它的子结点。当前结点访问结束后,递归函数将自动回到他的父结点。因此我们在函数退出之前要在路径上删除当前结点并减去当前结点的值,以确保返回父结点时,路径刚好是从根结点到父结点的路径。不难看出保存数据的路径结构实际上是一个栈,因为路径要与递归调用状态一致,而递归调用的本质就是一个压栈和出栈的过程。