算法-重建二叉树

题目: 输入某二叉树的前序遍历与中序遍历结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果均无重复数字,前序遍历序列为{},中序遍历序列为{},则重建出图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 条评论
登录 后参与评论

相关文章

来自专栏数据结构与算法

3143 二叉树的序遍历

3143 二叉树的序遍历 时间限制: 1 s 空间限制: 32000 KB 题目等级 : 白银 Silver 题目描述 Description 求...

2685
来自专栏从流域到海域

校招必考:根据二叉树遍历序列确定二叉树

根据二叉树的前序遍历和中序遍历求其后序遍历或者根据二叉树的后序遍历和中序遍历求其前序遍历是腾讯等校招的必考题,下面我们就来分析一下解题思路。 这道题本质上...

1835
来自专栏小二的折腾日记

牛客网-剑指offer-2

二叉树是觉得很烦的东西了,比链表复杂很多,看着头都有点疼啊,但是没办法,生活就是这样,只有把不会的会了才会进步,怕的变得不怕才能越来越厉害。 常规的理解一下:二...

792
来自专栏个人分享

二叉树的深度

输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。

712
来自专栏221-B

数据结构简单要点总结(转)

栈是只能在一端进行插入和删除的线性表。 (别看只是个定义,非常重要,已经道出了运算方法:只能在一端插入和删除。)

321
来自专栏计算机视觉与深度学习基础

Leetcode 164 Maximum Gap 桶排序好题

Given an unsorted array, find the maximum difference between the successive ele...

1915
来自专栏Android机器圈

排序算法

上面结果可以说明,虽然也是比较了和冒泡一样多的次数,但是交换缺少了很多。所以时间为N²/2

1745
来自专栏C/C++基础

求二叉树的深度和宽度

题目: 输入一个二叉树的根节点,求该树的深度。从根节点到叶子节点依次经过的节点(含根、叶节点)形成树的一条路径,最长路径的长度包含的节点数为为树的深度,即二...

972
来自专栏算法修养

HOJ 2156 &POJ 2978 Colored stones(线性动规)

Colored stones Time Limit: 1000MS Memory Limit: 65536K Total Submission...

3058
来自专栏一英里广度一英寸深度的学习

三路快排算法-求中位数问题(4)

step1排列数组的时间复杂度是O(N),空间复杂度是O(1) step2 递归调用的复杂度O(logN)

772

扫码关注云+社区