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

最新推荐文章于 2024-11-04 15:37:16 发布

tqyou85

最新推荐文章于 2024-11-04 15:37:16 发布

阅读量427

点赞数

分类专栏： c语言文章标签： tree null 数据结构 struct table 百度

本文链接：https://blog.csdn.net/tqyou85/article/details/4394029

版权

c语言专栏收录该内容

11 篇文章 0 订阅

订阅专栏

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

分析：这是百度的一道笔试题，考查对树这种基本数据结构以及递归函数的理解。
当访问到某一结点时，把该结点添加到路径上，并累加当前结点的值。如果当前结点为叶结点并且当前路径的和刚好等于输入的整数，则当前的路径符合要求，我们把它打印出来。如果当前结点不是叶结点，则继续访问它的子结点。当前结点访问结束后，递归函数将自动回到父结点。因此我们在函数退出之前要在路径上删除当前结点并减去当前结点的值，以确保返回父结点时路径刚好是根结点到父结点的路径。我们不难看出保存路径的数据结构实际上是一个栈结构，因为路径要与递归调用状态一致，而递归调用本质就是一个压栈和出栈的过程。
参考代码：

#include <stdio.h> #include <stdlib.h> #include <string.h> typedef struct tree{ int data; struct tree *lchild,*rchild; }TREE; typedef struct stack{ TREE *tree; struct stack *next; }STACK; void CreateTree(TREE **T,int n){ if((*T) == NULL){ (*T)=(TREE *)malloc(sizeof(TREE)); (*T)->data=n; (*T)->lchild=NULL; (*T)->rchild=NULL; } else if((*T)->data == n) return; else if((*T)->data > n) CreateTree(&((*T)->lchild),n); else CreateTree(&((*T)->rchild),n); return; } void push_stack(TREE *T,STACK *Q){ if(T == NULL){ printf("tree is null/n"); return; } STACK *q=Q; while(q->next != NULL) q=q->next; STACK *tmp=(STACK *)malloc(sizeof(STACK)); tmp->tree=T; tmp->next=NULL; q->next=tmp; } TREE *pop_stack(STACK *Q){ STACK *q=Q; if(q->next == NULL) return NULL; while(q->next->next != NULL) q=q->next; TREE *T=NULL; T=q->next->tree; free(q->next); q->next=NULL; return T; } void findPath(TREE *T,int num,STACK *Q){ if(T == NULL) return; static int sum=0; sum+=T->data; push_stack(T,Q); int isleaf=0; isleaf=((!(T->lchild)) && ((!(T->rchild)))); if(isleaf && sum==num){ STACK *p=Q; while(p->next != NULL){ p=p->next; printf("%d/t",p->tree->data); } printf("/n"); } if(T->lchild) findPath(T->lchild,num,Q); if(T->rchild) findPath(T->rchild,num,Q); sum-=T->data; pop_stack(Q); return; } int main(){ int a[5]={10,5,12,4,7}; int i; TREE *T=NULL; for(i=0;i<5;i++) CreateTree(&T,a[i]); STACK *Q=(STACK *)malloc(sizeof(STACK)); Q->next=NULL; printf("findPath:/n"); int num=22; findPath(T,num,Q); free(Q); Q=NULL; return 0; }