剑指offer：重建二叉树

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


分析：前序遍历的第一个元素就是根节点的值
  1. BinaryTreeNode* Construct(int* preorder, int* inorder, int length)
  2. {
  3.     if(preorder == NULL || inorder == NULL || length <= 0)
  4.         return NULL;
  6.     return ConstructCore(preorder, preorder + length - 1,
  7.         inorder, inorder + length - 1);
  8. }
  10. BinaryTreeNode* ConstructCore
  11. (
  12.     int* startPreorder, int* endPreorder, 
  13.     int* startInorder, int* endInorder
  14. )
  15. {
  16.     // 前序遍历序列的第一个数字是根结点的值
  17.     int rootValue = startPreorder[0];
  18.     BinaryTreeNode* root = new BinaryTreeNode();
  19.     root->m_nValue = rootValue;
  20.     root->m_pLeft = root->m_pRight = NULL;
  22.     if(startPreorder == endPreorder)
  23.     {
  24.         if(startInorder == endInorder && *startPreorder == *startInorder)
  25.             return root;
  26.         else
  27.             throw std::exception("Invalid input.");
  28.     }
  30.     // 在中序遍历中找到根结点的值
  31.     int* rootInorder = startInorder;
  32.     while(rootInorder <= endInorder && *rootInorder != rootValue)
  33.         ++ rootInorder;
  35.     if(rootInorder == endInorder && *rootInorder != rootValue)
  36.         throw std::exception("Invalid input.");
  38.     int leftLength = rootInorder - startInorder;
  39.     int* leftPreorderEnd = startPreorder + leftLength;
  40.     if(leftLength > 0)
  41.     {
  42.         // 构建左子树
  43.         root->m_pLeft = ConstructCore(startPreorder + 1, leftPreorderEnd, 
  44.             startInorder, rootInorder - 1);
  45.     }
  46.     if(leftLength < endPreorder - startPreorder)
  47.     {
  48.         // 构建右子树
  49.         root->m_pRight = ConstructCore(leftPreorderEnd + 1, endPreorder,
  50.             rootInorder + 1, endInorder);
  51.     }
  53.     return root;
  54. }





posted @ 2015-07-21 10:02  Lucas_1993  阅读(306)  评论(0)  编辑  收藏  举报