【数据结构】二叉树的重构

问题描述：已知二叉树的先序和中序序列，然后还原该二叉树。

（1）首先通过先序遍历找到首元素即为根节点。

（2） 再通过中序遍历找到根节点的左边序列，即为左子树，根节点的右边即为右子树。

（3）再对二叉树的左子树和右子树进行递归，递归到叶子结点，递归结束。

struct TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> in) {
 
        int in_size = in.size();
 
        if(in_size == 0)
 
            return NULL;
 
        vector<int> pre_left, pre_right, in_left, in_right;
 
        int val = pre[0];
 
        TreeNode* node = new TreeNode(val);//根节点进行初始化，内存分配 
        int p = 0;
 
        for(p; p < in_size; ++p){
 
            if(in[p] == val) //找到中序的根结点位置，用来划分做右子树
                break;
 
        }
 
        for(int i = 0; i < in_size; ++i){
 
            if(i < p){
 
                in_left.push_back(in[i]);//构造中序 
                pre_left.push_back(pre[i+1]);//构造先序
 
            }
 
            else if(i > p){
 
                in_right.push_back(in[i]);//构造中序
 
                pre_right.push_back(pre[i]);//构造先序
 
            }
 
        }
 
        node->left = reConstructBinaryTree(pre_left, in_left);
 
        node->right = reConstructBinaryTree(pre_right, in_right);
 
        return node;
 
    }

简而言之，其实上边代码主要是做划分，一层层的递归，其实就是先划分，然后找到每一层的根结点，直到递归到叶子结点，递归结束。