在二元树中找出和为某一值的所有路径

来源:互联网 发布:锁屏解锁软件 编辑:程序博客网 时间:2024/06/05 20:23

百度面试题目:

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

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

    分析:看到该题目的第一反应是:递归+回溯。首先当然要保存搜索的路径,并记录当前路径上所有元素的和。如果累积和与当前节点值的和大于输入的整数data,则将不进行加法运算,直接回退;如果与当前节点值的和刚好等于sum,且当前节点为叶子节点,则打印该路径,然后回退;如果与当前节点值得和小于data,则继续向下计算……,这样的做法似乎很合理,其实呢?很显然,如果这样做,我们就为问题添加了一个条件:即二元树上的元素值均为正数!!!题目显然没有给出这个条件。自然算法就错了……

    正确的做法应该是这样的:首先当然仍要保存搜索的路径,并记录当前路径上所有元素的和sum。如果当前节点为叶子节点,并且当前节点值与sum的和等于data,则满足条件,打印后递归返回到父节点,注意在打印后、递归返回之前要先减去当前节点元素的值;如果当前节点不是叶子节点,则不必判断当前节点值与sum的和是否等于data,继续访问子节点……

    另一种类似的方法不用录当前路径上所有元素的和sum,而是使用期望的和依次减去访问到的节点的值……最后是判断到达叶子节点时期望和是否减为0,其实时类似的……看代码实现:

#include <stdafx.h>#include<stdlib.h>#define MAX 20struct BiTreeNode{int data;struct BiTreeNode *left;struct BiTreeNode *right;};/*递归创建二叉排序树,以'-1'结束*/BiTreeNode * CreateBSTree(int *data,int pos,int len){BiTreeNode * tr;if(pos>=len){return NULL;}else{tr = (BiTreeNode *)malloc(sizeof(BiTreeNode));tr->data=data[pos];tr->left = CreateBSTree(data,2*pos+1,len);tr->right = CreateBSTree(data,2*pos+2,len);  return tr;}}//中序遍历二叉树void InOrderTraverse(BiTreeNode * root){if(root!=NULL){InOrderTraverse(root->left);printf("%d  ",root->data);InOrderTraverse(root->right);}}//打印路径void printPath(int path[],int top){printf("第一组:");for(int i=0;i<top;i++)printf("%d ",path[i]);printf("\n");}//查找和为sum的路径,path数组存放路径的值,top代表每个可行的路径中的元素个数void findPath(BiTreeNode *root, int sum,int top,int path[]){path[top++] = root->data;sum -= root->data;if (root->left == NULL && root->right==NULL) {if (sum == 0) {printPath(path, top);}} else {if (root->left != NULL) findPath(root->left, sum, top,path);if (root->right!=NULL) findPath(root->right, sum, top,path);}top --;sum += root->data;}int main(){int data[]={1,2,3,1,2,3,2,1,3,2,1,2,1,2,1};int len=sizeof(data)/sizeof(int);BiTreeNode * root=CreateBSTree(data,0,len);InOrderTraverse(root);printf("\n");int sum=6;int top=0;int path[MAX]={0};findPath(root,sum,top,path);return 0;}

 

原创粉丝点击