剑指Offer 通过中序和先序遍历重建二叉树

题目描述

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

 

思路：

写一个reConstruct函数4个形参，分别记录子树前序开始结束的位置，中序开始结束的位子。每次在中序中将前序的根节点找出，讲中序分为前（左子树），后（右子树）2 个部分。递归，直到，子树的开始位置大于结束位置。

 

 

AC代码：

 /**

  * Definition for binary tree

  * struct TreeNode {

  *     int val;

  *     TreeNode *left;

  *     TreeNode *right;

  *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}

  * };

  */

 class Solution {

 public:

     struct TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> in) {

         m_pre=pre;

         m_in=in;

         TreeNode* head;

         head=reConstruct(,pre.size()-,,in.size()-);

         return head;

     }

     struct TreeNode* reConstruct(int pre_L,int pre_R,int in_L,int in_R)

     {

         if(pre_L>pre_R||in_L>in_R)

             return nullptr;

         TreeNode *root = new TreeNode(m_pre[pre_L]);

         int i=;

         while(m_pre[pre_L]!=m_in[i])

             i++;

         root->left=reConstruct(pre_L+,pre_L+i-in_L,in_L,i-);

         root->right=reConstruct(i+pre_L+-in_L,pre_R,i+,in_R);

         return root;

     }

 private:

        //代替全局变量

     vector<int> m_pre;

     vector<int> m_in;

 };