重建二叉树

题目描述

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

先序（也就是先根），中序（也就是中跟）。

因为先序遍历的操作是：先根节点遍历，然后遍历右子树节点；中序遍历的操作是：先遍历左子树，然后根节点，然后右子树节点。根据这种特点，我们指知道，先序遍历的第一个节点即为该树的根节点，在中序遍历中，位于该数字之前的位于该树的左部分，相反，位于该树的右部分。所以，我们可以进行划分左右子树的方法进行递归操作。

/**
 * 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) {
    TreeNode *head;
if (!pre.empty())
head = new TreeNode(*(pre.begin()));
else
{
head = NULL;
return head;
}
//首先划分左右子树区间；
vector<int>left;//左子树区间
vector<int>right;//右子树区间
vector<int>prel;//左子树区间的先序遍历
vector<int>prer;//右子树区间的先序遍历
auto it = vin.begin();
int i = 1;
while (*it != *(pre.begin()))
{
prel.push_back(pre[i]);
i++;
left.push_back(*it);
it++;
}
it++;
while (it != vin.end())
{
prer.push_back(pre[i]);
i++;
right.push_back(*it);
it++;
}


if (!left.empty())
head->left = reConstructBinaryTree(prel, left);
if (!right.empty())
head->right = reConstructBinaryTree(prer, right);


return head;
    }
};