C++练习 | 递归创建二叉树并求叶子结点的数值和

最新推荐文章于 2023-07-04 00:02:16 发布

weixin_30624825

最新推荐文章于 2023-07-04 00:02:16 发布

阅读量355

点赞数

CC 4.0 BY-SA版权

文章标签：数据结构与算法 c/c++

原文链接：http://www.cnblogs.com/tsj816523/p/10619163.html

本文介绍了一种根据先序序列和中序序列构建二叉树的方法，并提供了一个递归算法来创建二叉树。此外，还实现了一个计算二叉树所有叶子节点之和的功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <iostream>
using namespace std;

struct Tree
{
    int data;
    Tree *lchild;
    Tree *rchild;
}tree;

Tree *Create(int a1[],int b1[],int n)
{
    int k;
    if(n<=0)
        return NULL;
    int root=a1[0];
    Tree *bt=(Tree *)malloc(sizeof(Tree));
    bt->data=root;
    for(k=0;k<n;k++)
    {
        if(b1[k]==root)
            break;//分割左右子树
    }
    bt->lchild=Create(a1+1,b1,k);
    bt->rchild=Create(a1+k+1,b1+k+1,n-k-1);
    return bt;
}

int Tsum(Tree *r)
{
    if(r->lchild==NULL&&r->rchild==NULL)
            return r->data;
    if(r->lchild==NULL&&r->rchild!=NULL)
        return Tsum(r->rchild);
    if(r->lchild!=NULL&&r->rchild==NULL)
        return Tsum(r->lchild);
    if(r->lchild!=NULL&&r->rchild!=NULL)
        return Tsum(r->lchild)+Tsum(r->rchild);
    return r->data;
}

int main()
{
    Tree *rt;
    int sum1;
    int a[107],b[107],N;//N为总节点数，a为先序序列，b为中序序列
    cin>>N;
    for(int i=0;i<N;i++)
        cin>>a[i];
    for(int i=0;i<N;i++)
        cin>>b[i];
    rt=Create(a,b,N);//存树
    sum1=Tsum(rt);
    cout<<sum1<<endl;
    return 0;
}