剑指Offer(第2版)——面试题7：重建二叉树

题目：

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

解题思路

知识点：
　　通常树有如下几种遍历方式：
　　前序序列：先访问根节点，再访问左子节点，最后访问右子节点。
　　中序序列：先访问左子节点，再访问根节点，最后访问右子节点。
　　后序序列：先访问左子节点，再访问右子节点，最后访问根节点。

• 前序遍历的第一个数字总是树的根节点的值。
• 中序遍历序列中，根节点的值在序列的中间，左子树的节点的值位于根节点的值的左边，而右子树的节点的值位于根节点的值的右边。
• 同理我们可以用递归的方法来重复这个过程。
  
  具体实施：
• 我们可以自己写一个函数，在传入两个序列的同时传入两个序列的起止位置下标(主要用来用来控制我们构建左右子树的范围)，当某一个序列的起始位置大于终止位置时返回nullptr；
• 在函数内部，首先用前序遍历第一个数字来给创建的根节点赋值；
• 接着在中序遍历序列中寻找与该数字相同的位置即为中序遍历中根节点的位置，在其左右分别是左右子树的中序序列；
• 此时我们就可以将问题分解为重建两个小二叉树，利用递归的方法不断深入来逐步构建二叉树的每一个节点，实现二叉树的重建。

代码

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        // 如果前序遍历或中序遍历数组为空，直接返回nullptr
        if(pre.size() == 0 || vin.size() == 0){
            return nullptr;
        }           
        // 重建二叉树函数
        TreeNode* ret = reConstructBinaryTree1(pre,0,pre.size()-1,vin,0,vin.size()-1);
        return ret;
    }

public:
    TreeNode* reConstructBinaryTree1(vector<int> pre,int p1,int p2,vector<int> vin,int v1,int v2)
    {
        // 当前序或中序遍历起始位置大于终止位置时，返回nullptr
        if(p1>p2 || v1>v2){
            return nullptr;
        }           
        // 定义根节点，开辟空间并用前序遍历第一个值来赋值
        TreeNode* root = new TreeNode(pre[p1]);
        
        // 在中序遍历中循环查找根节点位置，递归构建左右子树的根节点
        for(int i=v1;i<=v2;++i)
        {
            if(pre[p1] == vin[i])
            {
                // 找到根节点后构建左子树，传入前序遍历中左子树(根节点后i-v1个元素)的起止位置和中序遍历中左子树(根节点左边数组)的起止位置
                root->left = reConstructBinaryTree1(pre,p1+1,p1+i-v1,vin,v1,i-1);
                // 找到根节点后构建右子树，传入前序遍历中右子树(根节点后i-v1+1开始到终止位置)的起止位置和中序遍历中右子树(根节点右边数组)的起止位置
                root->right = reConstructBinaryTree1(pre,p1+i-v1+1,p2,vin,i+1,v2);
            }
        }       
        return root;
    }
};