Cracking the coding interview--Q4.8

题目

原文：

You are given a binary tree in which each node contains a value. Design an algorithm to print all paths which sum up to that value. Note that it can be any path in the tree - it does not have to start at the root.

译文：

给一个每个结点都包含一个值的二叉树，设计一个算法打印所有满足这个条件的路径：路径上结点的值加起来等于给定的一个值。注意：这些路径不一定从根节点开始。

解答

结点中包含指向父亲结点的指针，那么，只需要去遍历这棵二叉树， 然后从每个结点开始，不断地去累加上它父亲结点的值直到父亲结点为空(这个具有唯一性， 因为每个结点都只有一个父亲结点。也正因为这个唯一性， 可以不另外开额外的空间来保存路径)，如果等于给定的值sum，则打印输出。

代码如下：

import java.util.Stack;

class Q4_8{
	public static void findSum(TreeNode head,int sum){
		if(head==null) return;
		TreeNode tnode=head;
		int tmp=0;
		for(int i=1;tnode!=null;i++){
			tmp+=tnode.value;
			if(tmp==sum)
				print(head,i);
			tnode=tnode.parent;
		}
		findSum(head.lchild,sum);
		findSum(head.rchild,sum);
	}
	private static void print(TreeNode tnode,int level){
		Stack<Integer> s=new Stack<Integer>();
		for(int i=0;i<level;i++){
			s.push(tnode.value);
			tnode=tnode.parent;
		}
		while(s.size()>0){
			System.out.print(s.pop()+" ");
		}
	}
	public static void main(String[] args){
		int[] arr={5,1,3,8,6,10};
		TreeNode root=TreeNode.createBinaryTree(arr);
		findSum(root,23);
	}
}

---EOF---