[剑指Offer]07. 重建二叉树

为了面试做准备！开始刷剑指Offer，今天做的值得记录的是07. 重建二叉树，题意如下：

我们知道前序遍历的第一个是根节点，而中序遍历从根节点处把序列分为了左子树，根节点，右子树。所以我们可以根据前序遍历找根节点，根据中序遍历划分子树。使用递归的方法把大问题转化成为小问题。具体还有很多小细节，见代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
    private:
      vector<int> pre;
      map<int,int> mp;//用来记录根节点在中序遍历的位置，优化时间复杂度。
      TreeNode* retree(int root,int left,int right)
      {
          if(left>right)
          return nullptr;
          TreeNode* now = new TreeNode(pre[root]);//新建节点   
          int n=mp[pre[root]];
          now->left=retree(root+1,left,n-1);
          //划分左子树，左子树根节位置为根节点加一，左边界还是left，右边界为根节点之前的值。
          now->right=retree(root+n-left+1,n+1,right);
          //划分右子树，右子树的根节点为根节点之后的左节点的个数个，右边界不变
          return now;
      }
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
      this->pre=preorder;
      for(int i=0;i<inorder.size();i++)
      {
          mp[inorder[i]]=i;
      }
      return retree(0,0,inorder.size()-1);
    }
};

时间复杂度O(N)，空间复杂度O(N)。