欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

【剑指offer】重建二叉树

程序员文章站 2022-06-17 18:24:19
...

题目:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如:前序遍历序列{ 1, 2, 4, 7, 3, 5, 6, 8}和中序遍历序列{4, 7, 2, 1, 5, 3, 8,6},重建出下图所示的二叉树并输出它的头结点。

【剑指offer】重建二叉树

二叉树结点的定义如下:

struct BinaryTreeNode
{
    int m_nValue;
    BinaryTreeNode * m_pLeft;
    BinaryTreeNode * m_pRight;
};

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

    BinaryTreeNode * pNode = new BinaryTreeNode();
    int tmpLength = 0;
    int * tmpPreorder = preorder, * tmpInorder = inorder;

    // 初始化 pNode
    pNode->m_nValue = preorder[0];
    pNode->m_pLeft = pNode->m_pRight = NULL;
    
    while( *tmpInorder++ != tmpPreorder[0] ) // 计算左子树有多少个节点
        tmpLength++;

    if( tmpLength > 0 ) // 构建左子树
        pNode->m_pLeft = construct( preorder + 1, inorder, tmpLength );
        
    if( length - tmpLength - 1 > 0 ) // 构建右子树
        pNode->m_pRight = construct( preorder + tmpLength + 1, inorder + tmpLength + 1, length - tmpLength - 1 );
        
    return pNode;
}