算法-重建二叉树

题目: 输入某二叉树的前序遍历与中序遍历结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果均无重复数字,前序遍历序列为{},中序遍历序列为{},则重建出图2.6所示的二叉树并输出他的头结点。二叉树的结点定义如下:

struct BiTNode
{  
    int data;  
    BiTNode* lchild;  
    BiTNode* rchild;  
}; 

解题思路:

二叉树遍历总结中,我们介绍了常用的遍历方法,那么前序遍历序列的第一个点一定是根节点,又由于无重复数字,所以我们就知道了根结点在中序遍历中的位置:

又由于中序遍历的特点,根结点的左子结点一定是{4,7,2},右子结点{5,3,8,6}。接下来我们再把左右子结点想象成一个单独的根结点,那么,他们又有了自己的根结点:

截止到这里,我们能确定位置的点就是1,2,3,以及它们的左右子结点(具体位置不清楚):

下面的步骤就是,继续根据前序确定根结点,根据中序确定左右子结点,直到确定了所有的位置:

代码实现: 上面说到,每一次确定一个根结点和对应的左右子结点,然后在把左右子结点想象单独的树,重复之前的步骤,显然这是个递归:

BinaryTreeNode* Construct(int* preorder, int* inorder, int length)
{
    if(preorder == NULL || inorder == NULL || length <= 0)
        return NULL;

    return ConstructCore(preorder, preorder + length - 1,
        inorder, inorder + length - 1);
}

BinaryTreeNode* ConstructCore(int* startPreorder, int* endPreorder, int* startInorder, int* endInorder)
{
    // 前序遍历序列的第一个数字是根结点的值
    int rootValue = startPreorder[0];
    BinaryTreeNode* root = new BinaryTreeNode();
    root->m_nValue = rootValue;
    root->m_pLeft = root->m_pRight = NULL;

    if(startPreorder == endPreorder)
    {
        if(startInorder == endInorder && *startPreorder == *startInorder)
            return root;
        else
            throw std::exception("Invalid input.");
    }

    // 在中序遍历中找到根结点的值
    int* rootInorder = startInorder;
    while(rootInorder <= endInorder && *rootInorder != rootValue)
        ++ rootInorder;

    if(rootInorder == endInorder && *rootInorder != rootValue)
        throw std::exception("Invalid input.");

    int leftLength = rootInorder - startInorder;
    int* leftPreorderEnd = startPreorder + leftLength;
    if(leftLength > 0)
    {
        // 构建左子树
        root->m_pLeft = ConstructCore(startPreorder + 1, leftPreorderEnd, 
            startInorder, rootInorder - 1);
    }
    if(leftLength < endPreorder - startPreorder)
    {
        // 构建右子树
        root->m_pRight = ConstructCore(leftPreorderEnd + 1, endPreorder,
            rootInorder + 1, endInorder);
    }

    return root;
}

可以看到相比于链表,树相关的代码明显更多了,重建二叉树要50行左右。 代码由两个函数组成,Construct函数的输入参数是int*型,这是因为前序与中序遍历序列是int型的数组:

 int preorder[length] = {1, 2, 4, 7, 3, 5, 6, 8};
 int inorder[length] = {4, 7, 2, 1, 5, 3, 8, 6};

重建的过程都在ConstructCore函数中,输入前序中序的开始于结束为止的地址,每一次都把前序序列的开始的地址作为新的树的根结点:

int rootValue = startPreorder[0];

递归退出的条件,就是前序序列开始的点与结束的点是一个。

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 【腾讯云的1001种玩法】构建企业级应用环境之数据层面优化(一)

    本系列为两年前 VMCloud 云平台的进阶篇,本次借助 QCloud 的《1001种玩法》活动来继续完成进阶篇,主要以在 QCloud 上搭建一个完整的应用环...

    李斯达
  • Amazon Aurora:云时代的数据库 ( 中)

    文章详尽的介绍了 Aurora 设计背后的驱动和思考,以及如何在云上实现一个同时满足高并发、高吞吐量、高稳定性、高可用、高扩展的云数据库。

    谭伟华)
  • Spider 引擎分布式数据库解决方案(最全的 spider 教程)

    最近开始负责财付通的数据库的相关维护工作,其中有几套系统使用的 spider 引擎,本文将 spider 引擎的功能、使用场景、部署、实战测试等做个简单的总结。

    飞鸿无痕
  • 基于 MySQL 5.7多源复制及 Keepalived 搭建三节点高可用架构

    MySQL 5.7发布后,在复制方面有了很大的改进和提升。比如开始支持多源复制以及真正的支持多线程复制。本文讲述基于 MySQL 5.7多源复制及 Keepal...

    叶金荣
  • 分区表场景下的 SQL 优化

    有个表做了分区,每天一个分区。该表上有个查询,经常只查询表中某一天数据,但每次都几乎要扫描整个分区的所有数据,有什么办法进行优化吗?

    叶金荣
  • 腾讯云分布式数据库(DCDB)

    DCDB 是部署在腾讯云公有云上的一种兼容MySQL协议和语法,支持自动水平拆分的share nothing架构的分布式数据库。分布式数据库即业务获取是完整的逻...

    苏强
  • Scrapyd 日志输出定时清理

    现在维护着一个新浪微博爬虫,爬取量已经5亿+,使用了Scrapyd部署分布式。Scrapyd运行时会输出日志到本地,导致日志文件会越来越大,几十个G,这个其实就...

    崔庆才
  • 腾讯云采购节服务器三折起,这才是人民的云计算!

    4月24日,腾讯云正式启动 “腾讯云采购节”大促活动,活动期间旗下云服务器、云数据库、域名等多款核心产品大幅降价,最低折扣达3折,优惠方式也涵盖了新用户礼包、大...

    云资讯小编
  • 公有云的商业模式分析

    最近看到一篇文章,里面以酒店为比喻,对公有云的商业模式进行了分析,还比较生动、形象。在社区里摘录如下。

    EarlGrey
  • Amazon Aurora:云时代的数据库 ( 上)

    文章是 Amazon 在 SIGMOD\'17 上最新发表的关于 Aurora 论文的翻译版本,详尽的介绍了Aurora 设计背后的驱动和思考,以及如何在云上实...

    谭伟华)

扫码关注云+社区

领取腾讯云代金券