剑指offer---重建二叉树(6)

来源:互联网 发布:英雄联盟t恤淘宝 编辑:程序博客网 时间:2024/06/05 16:59


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

二叉树的定义如下:

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


利用递归建立左右二叉树。
具体思想在此不做详细介绍。详情请看相关资料。

#include <iostream>using namespace std;struct BinaryTreeNode{   int  m_nValue;   BinaryTreeNode*   m_pLeft;   BinaryTreeNode*   m_pRight;};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 = NULL;    root->m_pRight = NULL;    if(startPreorder == endPreorder)    {        if(startInorder == endInorder && *startPreorder == *endPreorder)        {            return root;        }        else        {            throw exception("Invalid input");        }    }    //在中序遍历中找到根结点的值    int* rootInorder = startInorder;    while(rootInorder <= endInorder && *rootInorder != rootValue)        rootInorder++;    if(rootInorder == endInorder && *rootInorder != rootValue)    {        throw exception("Invalid input");    }    int leftlengt = rootInorder-startInorder;    int* leftPreorder = startPreorder + leftlengt;    if(leftlengt > 0)    {        root->m_pLeft = ConstructCore(startPreorder +1,leftPreorder,startInorder,rootInorder-1);    }    if(leftlengt < endPreorder-startPreorder)    {        root->m_pRight = ConstructCore(leftPreorder+1,endPreorder,rootInorder+1,endInorder);    }    return root;}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+1);}
原创粉丝点击