二叉树中根到叶子路径中节点和等于给定值的路径

1. 描述

输入一个整数和一棵二元树。从树的根结点开始往下访问一直到叶结点所经过的所有结点形成一条
路径。打印出和与输入整数相等的所有路径。
例如输入整数 22 和如下二元树
                    10
                   /    \
                 5      12

               /    \ 
             4      7
则打印出两条路径:10, 12 和 10, 5, 7。

据说这是百度的一道笔试题。

结点的存储结构见<二叉树的操作>中存储结构部分。

算法如下：

[cpp] view plaincopyprint?

1. #define N 50  
2. BiNode *stack[N]; //定义栈存储路径上的结点指针  
3. int top = 0;  
4.   
5. void findPath(BiNode *T, int sum, int currSum)   
6. {  
7.     if (! T) return;  
8.     currSum += T->data;  
9.     stack[top++] = T;  
10.     if (! T->lchild && ! T->rchild && currSum == sum) {  
11.         int i;  
12.         for (i = 0; i < top; i++)  
13.             printf("%d ", stack[i]->data); //打印路径节点的值  
14.         printf("\n");  
15.     } else {  
16.         findPath(T->lchild, sum, currSum);  
17.         findPath(T->rchild, sum, currSum);  
18.     }  
19.     currSum -= T->data;  
20.     top--;  
21. }  

sum为给定的值，初始调用currSum为零。

代码如下：

[cpp] view plaincopyprint?

1. #include "headfiles/BiNode.h"  
2. #include "sharedSource/createBi_pre_in.c"  
3.   
4. void findPath(BiNode *T, int sum, int currSum);  
5. int main(void)  
6. {  
7.     BiNode *T;  
8.     char pre[] = {5,4,6,3,2,7,4,1,2};  
9.     char in[] =  {3,6,2,4,5,1,4,7,2};  
10.     T = createBi_pre_in(pre, in, 0,0, 9);  
11.     findPath(T, 17, 0);  
12.   
13.     return 0;  
14. }